Update SiegeManager.java

APDevTeam · Jul 2, 2024 · 68eb45c · 68eb45c
1 parent b800a27
commit 68eb45c
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/src/main/java/net/countercraft/movecraft/warfare/features/siege/SiegeManager.java b/src/main/java/net/countercraft/movecraft/warfare/features/siege/SiegeManager.java
@@ -66,14 +66,14 @@ public void run() {
         }
 
         for (Siege siege : sieges) {
-            if (siege.getStage().get() == Siege.Stage.IN_PROGRESS) {
-                new SiegeProgressTask(siege).runTask(warfare);
-            }
-            else if (siege.getStage().get() == Siege.Stage.PREPARATION) {
-                new SiegePreparationTask(siege).runTask(warfare);
-            }
-            else {
-                // Siege is inactive, do nothing
+            switch (siege.getStage().get()) {
+                case IN_PROGRESS:
+                case SUDDEN_DEATH:
+                    new SiegeProgressTask(siege).runTask(warfare);
+                    break;
+                case PREPARATION:
+                    new SiegePreparationTask(siege).runTask(warfare);
+                    break;
             }
         }
     }