diff --git a/OGP1718-Worms/src/worms/model/Team.java b/OGP1718-Worms/src/worms/model/Team.java index 9f7c571..2447ce4 100644 --- a/OGP1718-Worms/src/worms/model/Team.java +++ b/OGP1718-Worms/src/worms/model/Team.java @@ -58,10 +58,10 @@ public class Team { if (!canHaveAsWorm(worms)) throw new IllegalArgumentException(); Collection wormCollection = getAllWormsOfTeam(); - - Arrays.stream(worms).forEach(w -> w.setTeam(this)); - - getAllWormsOfTeam().addAll(Arrays.asList(worms)); + Arrays.stream(worms).forEach(w -> { + w.setTeam(this); + wormCollection.add(w); + }); } /**