diff --git a/Battleship/Battleship/SetGeneric.cs b/Battleship/Battleship/SetGeneric.cs index c0e3858..7542dea 100644 --- a/Battleship/Battleship/SetGeneric.cs +++ b/Battleship/Battleship/SetGeneric.cs @@ -72,7 +72,9 @@ namespace Battleship.Generics yield return _places[i]; if (maxShips.HasValue && i == maxShips.Value) { + yield break; + } } }