Fix newing some references for turn

SharifAIChallenge · Mar 3, 2020 · 420d59f · 420d59f
1 parent e3f378a
commit 420d59f
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/src/Client/Model/Map.java b/src/Client/Model/Map.java
@@ -27,7 +27,6 @@ private Map(int rows, int cols) {
     }
 
     public static Map createMap(int rows, int cols) {
-        if (map != null) return map;
         map = new Map(rows, cols);
         return map;
     }

diff --git a/src/Client/dto/turn/ClientTurnMessage.java b/src/Client/dto/turn/ClientTurnMessage.java
@@ -37,7 +37,7 @@ public class ClientTurnMessage {
     public ClientTurnMessage() { }
 
     private void updateMapUnits(TurnMessage turnMessage){
-        Map.getMap().getUnits().clear();
+        Map.getMap().setUnits(new ArrayList<>());
         for(Unit unit : turnMessage.getUnits())
             Map.getMap().getUnits().add(unit);
     }