fix(FishCalculator): use correct method for getting fish

oscar-schwarz · Jun 17, 2024 · 48f0c87 · 48f0c87
1 parent 441aae9
commit 48f0c87
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/java/fish_variety/FishCalculator.java b/src/main/java/fish_variety/FishCalculator.java
@@ -9,7 +9,7 @@ public static FishCollection mostCompatibleFishTypesOfCollection(int price, Fish
 
         ArrayList<FishCollection> variants = new ArrayList<>();
 
-        for (Fish fish : constraints.getAllFish()) {
+        for (Fish fish : initialCollection.getUnaddedFishes()) {
             FishCollection collectionVariant = new FishCollection(constraints, initialCollection.getFishes());
 
             collectionVariant.add(fish);
@@ -19,7 +19,7 @@ public static FishCollection mostCompatibleFishTypesOfCollection(int price, Fish
             }
         }
 
-        if (variants.size() == 0) {
+        if (variants.isEmpty()) {
             return initialCollection;
         }