From 1e2eddafc7e861bc1e9e573ca49acb017f6ee2aa Mon Sep 17 00:00:00 2001
From: Achim <achim.retzbach@outlook.de>
Date: Mon, 20 Jun 2016 18:15:20 +0200
Subject: [PATCH] ArmySchubser - Version 1.0.0

Rosen sind rot,
Veilchen sind blau,
der Schubser is jetzt feddisch,
ich weiss es genau.
---
 ArmySchubser.java | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/ArmySchubser.java b/ArmySchubser.java
index 307b18e..b05d1a3 100644
--- a/ArmySchubser.java
+++ b/ArmySchubser.java
@@ -6,7 +6,7 @@ import javax.swing.JOptionPane;
  * Schubst Einheiten umher.
  * 
  * @author MaxiJohl, GruenerWal
- * @version 0.3.0
+ * @version 1.0.0
  */
 
 public class ArmySchubser extends Map_World
@@ -33,15 +33,19 @@ public class ArmySchubser extends Map_World
         String toMoveString = JOptionPane.showInputDialog(null, "Wieviele Einheiten willst du verschieben?");
         Integer entitiesToMove = Integer.valueOf(toMoveString);
 
-        if ( (sourceProvince.getEntityCount() - entitiesToMove) > 0)
+        if ( (sourceProvince.getEntityCount() - entitiesToMove) > 0 && entitiesToMove != null)
         {
             sourceProvince.removeFromEntities(entitiesToMove);
             destinationProvince.addToEntities(entitiesToMove);
+            sourceProvince = null;
+            destinationProvince = null;
         }
 
-        if ( (sourceProvince.getEntityCount() - entitiesToMove) <= 0 )
+        if ( (sourceProvince.getEntityCount() - entitiesToMove) <= 0 && entitiesToMove != null)
         {
             JOptionPane.showMessageDialog(null,"Du hast nicht genügend Einheiten, um die gewünschte Anzahl von " + sourceProvince.getDisplayName() + " nach " + destinationProvince.getDisplayName() + " zu verschieben, Köhler.");
+            sourceProvince = null;
+            destinationProvince = null;
         }
     }
 
@@ -91,5 +95,4 @@ public class ArmySchubser extends Map_World
             }
         }
     }
-}
-
+}
\ No newline at end of file