diff --git a/Projekte/battleship/model/src/main/java/pp/battleship/game/client/BattleState.java b/Projekte/battleship/model/src/main/java/pp/battleship/game/client/BattleState.java index fa8d731d..0f27eb4f 100644 --- a/Projekte/battleship/model/src/main/java/pp/battleship/game/client/BattleState.java +++ b/Projekte/battleship/model/src/main/java/pp/battleship/game/client/BattleState.java @@ -9,6 +9,7 @@ import pp.battleship.message.client.ShootMessage; import pp.battleship.message.server.EffectMessage; +import pp.battleship.model.Battleship; import pp.battleship.model.IntPoint; import pp.battleship.model.ShipMap; import pp.battleship.notification.Sound; @@ -60,7 +61,9 @@ public void receivedEffect(EffectMessage msg) { if (destroyedOpponentShip(msg)) logic.getOpponentMap().add(msg.getDestroyedShip()); if (msg.isGameOver()) { - msg.getRemainingOpponentShips().forEach(logic.getOwnMap()::add); + for(Battleship ship: msg.getRemainingOpponentShips()){ + logic.getOpponentMap().add(ship); + } logic.setState(new GameOverState(logic)); } }