2 files changed, 2 insertions, 2 deletions
diff --git a/core/src/ch/asynk/tankontank/game/states/StateDeployment.java b/core/src/ch/asynk/tankontank/game/states/StateDeployment.java
index 3be9201..a30d59c 100644
--- a/core/src/ch/asynk/tankontank/game/states/StateDeployment.java
+++ b/core/src/ch/asynk/tankontank/game/states/StateDeployment.java
@@ -61,7 +61,7 @@ public class StateDeployment extends StateCommon
     @Override
     public void touchUp()
     {
-        Unit unit = ctrl.hud.playerInfo.getDockUnit();
+        Unit unit = ctrl.hud.playerInfo.unitDock.selectedUnit;
         if (!done && (unit != null) && (unit != activeUnit)) {
             changeUnit(unit);
         } else if (selectedUnit != null) {
diff --git a/core/src/ch/asynk/tankontank/game/states/StateReinforcement.java b/core/src/ch/asynk/tankontank/game/states/StateReinforcement.java
index 2c423c6..33668db 100644
--- a/core/src/ch/asynk/tankontank/game/states/StateReinforcement.java
+++ b/core/src/ch/asynk/tankontank/game/states/StateReinforcement.java
@@ -50,7 +50,7 @@ public class StateReinforcement extends StateCommon
     @Override
     public void touchUp()
     {
-        Unit unit = ctrl.hud.playerInfo.getDockUnit();
+        Unit unit = ctrl.hud.playerInfo.unitDock.selectedUnit;
         if ((unit != null) && (unit != activeUnit))
             changeUnit(unit);
         else if ((entryPoint != null) && upHex.isEmpty() && entryPoint.contains(upHex))