2 files changed, 4 insertions, 4 deletions
diff --git a/core/src/ch/asynk/tankontank/game/states/StateEngage.java b/core/src/ch/asynk/tankontank/game/states/StateEngage.java
index 4b1539b..0ef213d 100644
--- a/core/src/ch/asynk/tankontank/game/states/StateEngage.java
+++ b/core/src/ch/asynk/tankontank/game/states/StateEngage.java
@@ -19,7 +19,7 @@ public class StateEngage extends StateCommon
             activeUnit = null;
             // use selectedHex and selectedUnit
             map.hidePossibleTargets();
-            map.collectPossibleTargets(selectedUnit, ctrl.opponent.units);
+            map.collectPossibleTargets(selectedUnit, ctrl.opponent.unitsAsPawns());
             map.showPossibleTargets();
             if (to != null) {
                 // quick fire -> replay touchUp
@@ -89,7 +89,7 @@ public class StateEngage extends StateCommon
             to = upHex;
             activeUnit = unit;
             activeUnit.showTarget();
-            map.collectAttackAssists(selectedUnit, activeUnit, ctrl.player.units);
+            map.collectAttackAssists(selectedUnit, activeUnit, ctrl.player.unitsAsPawns());
             map.showAttackAssists();
             ctrl.hud.actionButtons.show(Buttons.ENGAGE.b | Buttons.DONE.b | ((ctrl.cfg.canCancel) ? Buttons.ABORT.b : 0));
         }
diff --git a/core/src/ch/asynk/tankontank/game/states/StateSelect.java b/core/src/ch/asynk/tankontank/game/states/StateSelect.java
index 7bdc815..ae70328 100644
--- a/core/src/ch/asynk/tankontank/game/states/StateSelect.java
+++ b/core/src/ch/asynk/tankontank/game/states/StateSelect.java
@@ -77,9 +77,9 @@ public class StateSelect extends StateCommon
             int moves = map.collectPossibleMoves(selectedUnit);
             int targets = 0;
             if (isEnemy)
-                targets = map.collectPossibleTargets(selectedUnit, ctrl.player.units);
+                targets = map.collectPossibleTargets(selectedUnit, ctrl.player.unitsAsPawns());
             else
-                targets = map.collectPossibleTargets(selectedUnit, ctrl.opponent.units);
+                targets = map.collectPossibleTargets(selectedUnit, ctrl.opponent.unitsAsPawns());
             if (moves > 0)
                 map.collectMoveablePawns(selectedUnit);
             showPossibilities(selectedUnit);