game.UnitSet is dead : welcome UnitList

2 files changed, 3 insertions, 3 deletions

diff --git a/core/src/ch/asynk/tankontank/game/states/StateDeployment.java b/core/src/ch/asynk/tankontank/game/states/StateDeployment.java
index ace1265..3a6f2a5 100644
--- a/core/src/ch/asynk/tankontank/game/states/StateDeployment.java
+++ b/core/src/ch/asynk/tankontank/game/states/StateDeployment.java
@@ -4,7 +4,7 @@ import ch.asynk.tankontank.engine.Orientation;
 import ch.asynk.tankontank.game.Hex;
 import ch.asynk.tankontank.game.Zone;
 import ch.asynk.tankontank.game.Unit;
-import ch.asynk.tankontank.game.UnitSet;
+import ch.asynk.tankontank.game.UnitList;
 import ch.asynk.tankontank.game.hud.ActionButtons.Buttons;
 
 import ch.asynk.tankontank.TankOnTank;
@@ -13,7 +13,7 @@ public class StateDeployment extends StateCommon
 {
     private boolean done;
     private Zone entryZone;
-    private UnitSet deployedUnits = new UnitSet(map, 10);
+    private UnitList deployedUnits = new UnitList(10);
 
     @Override
     public void enter(StateType prevState)
diff --git a/core/src/ch/asynk/tankontank/game/states/StateMove.java b/core/src/ch/asynk/tankontank/game/states/StateMove.java
index 13280ce..caeebb2 100644
--- a/core/src/ch/asynk/tankontank/game/states/StateMove.java
+++ b/core/src/ch/asynk/tankontank/game/states/StateMove.java
@@ -38,7 +38,7 @@ public class StateMove extends StateCommon
             if (selectedUnit.canMove()) {
                 changeUnit(selectedUnit);
             } else {
-                changeUnit((Unit) map.moveableUnits.first());
+                changeUnit((Unit) map.moveableUnits.get(0));
             }
         }
     }