1 files changed, 2 insertions, 1 deletions
diff --git a/core/src/ch/asynk/rustanddust/game/Ctrl.java b/core/src/ch/asynk/rustanddust/game/Ctrl.java
index 8e05dad..faa2bc7 100644
--- a/core/src/ch/asynk/rustanddust/game/Ctrl.java
+++ b/core/src/ch/asynk/rustanddust/game/Ctrl.java
@@ -6,6 +6,7 @@ import com.badlogic.gdx.math.Vector3;
 
 import ch.asynk.rustanddust.RustAndDust;
 import ch.asynk.rustanddust.ui.Position;
+import ch.asynk.rustanddust.game.Map.UnitType;
 import ch.asynk.rustanddust.game.State.StateType;
 import ch.asynk.rustanddust.game.states.StateSelect;
 import ch.asynk.rustanddust.game.states.StateMove;
@@ -165,7 +166,7 @@ public class Ctrl implements Disposable
 
         if (nextState == StateType.DONE) {
             map.actionDone();
-            if (map.activatedUnits.size() > 0) {
+            if (map.unitsSize(UnitType.ACTIVATED) > 0) {
                 RustAndDust.debug("Ctrl", "burn down 1AP");
                 hud.notify("1 Action Point burnt", 0.6f, Position.BOTTOM_CENTER, false);
                 player.burnDownOneAp();