From ae0e7d1c1c03eceeba145a47d96c87d87df32fc5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?J=C3=A9r=C3=A9my=20Zurcher?= <jeremy@asynk.ch>
Date: Tue, 20 Jan 2015 09:27:32 +0100
Subject: StateDeployment: fix undo

---
 core/src/ch/asynk/tankontank/game/states/StateDeployment.java | 7 +++++--
 1 file changed, 5 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 c3a333e..6129631 100644
--- a/core/src/ch/asynk/tankontank/game/states/StateDeployment.java
+++ b/core/src/ch/asynk/tankontank/game/states/StateDeployment.java
@@ -44,7 +44,8 @@ public class StateDeployment extends StateCommon
     @Override
     public StateType abort()
     {
-        undo();
+        if (activeUnit != null)
+            undo();
         return StateType.DEPLOYMENT;
     }
 
@@ -73,8 +74,10 @@ public class StateDeployment extends StateCommon
                 unitEnter(activeUnit);
         } else {
             unit = downHex.getUnit();
-            if (deployedUnits.contains(unit))
+            if (deployedUnits.contains(unit)) {
                 showRotation(unit, downHex);
+                activeUnit = unit;
+            }
         }
     }
 
-- 
cgit v1.1-2-g2b99