fix entity wrapper task causing NPE onReload (#468)

WeaponMechanics · Oct 12, 2024 · d8d6c8d · d8d6c8d
1 parent 6584d3a
commit d8d6c8d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/WeaponMechanics/src/main/java/me/deecaad/weaponmechanics/WeaponMechanics.java b/WeaponMechanics/src/main/java/me/deecaad/weaponmechanics/WeaponMechanics.java
@@ -617,7 +617,9 @@ public void onDisable() {
         // This won't cancel move tasks on Folia... We have to do it manually
         foliaScheduler.cancelTasks();
         for (EntityWrapper entityWrapper : entityWrappers.values()) {
-            entityWrapper.getMoveTask().cancel();
+            TaskImplementation<Void> moveTask = entityWrapper.getMoveTask();
+            if (moveTask != null)
+                moveTask.cancel();
         }
 
         BlockDamageData.regenerateAll();