diff --git a/src/main/java/eu/mhsl/craftattack/spawn/appliances/settings/Settings.java b/src/main/java/eu/mhsl/craftattack/spawn/appliances/settings/Settings.java index f1ba0d7..4703619 100644 --- a/src/main/java/eu/mhsl/craftattack/spawn/appliances/settings/Settings.java +++ b/src/main/java/eu/mhsl/craftattack/spawn/appliances/settings/Settings.java @@ -135,11 +135,15 @@ public class Settings extends Appliance { } private int calculateInvSize(List> settings) { - int countOfUncategorized = (int) settings.stream().filter(setting -> !(setting instanceof CategorizedSetting)).count(); + int countOfUncategorized = (int) settings.stream() + .filter(setting -> !(setting instanceof CategorizedSetting)) + .count(); + return Arrays.stream(SettingCategory.values()) .map(settingCategory -> settings.stream() .filter(setting -> setting instanceof CategorizedSetting) - .filter(setting -> ((CategorizedSetting) setting).category().equals(settingCategory)) + .map(setting -> (CategorizedSetting) setting) + .filter(categorizedSetting -> categorizedSetting.category().equals(settingCategory)) .count()) .map(itemCount -> itemCount + countOfUncategorized) .map(itemCount -> (int) Math.ceil((double) itemCount / 9))