Populate entityMaxCounts in LifecycleEvent.SETUP

Author: Discusser

common/src/main/java/io/github/discusser/toomanyentities/TooManyEntities.java

@@ -1,13 +1,15 @@
 package io.github.discusser.toomanyentities;
 
 import dev.architectury.event.events.client.ClientTickEvent;
+import dev.architectury.event.events.common.LifecycleEvent;
 import dev.architectury.injectables.annotations.ExpectPlatform;
 import io.github.discusser.toomanyentities.client.TooManyEntitiesKeys;
 import io.github.discusser.toomanyentities.config.MapGuiProvider;
 import io.github.discusser.toomanyentities.config.TooManyEntitiesConfig;
 import me.shedaniel.autoconfig.AutoConfig;
 import me.shedaniel.autoconfig.gui.registry.GuiRegistry;
 import me.shedaniel.autoconfig.serializer.GsonConfigSerializer;
+import net.minecraft.registry.Registries;
 import net.minecraft.text.Style;
 import net.minecraft.text.Text;
 import net.minecraft.util.Formatting;
@@ -30,6 +32,8 @@ public static void initClient() {
         registry.registerPredicateProvider(new MapGuiProvider(), field -> Map.class.isAssignableFrom(field.getType()));
         TooManyEntitiesConfig.instance = AutoConfig.getConfigHolder(TooManyEntitiesConfig.class).getConfig();
 
+        LifecycleEvent.SETUP.register(() -> Registries.ENTITY_TYPE.stream().forEach(entityType -> TooManyEntitiesConfig.instance.entityMaxCounts.put(entityType.getTranslationKey(), 0)));
+
         ClientTickEvent.CLIENT_POST.register(minecraft -> {
             while (TooManyEntitiesKeys.KEY_TOGGLE_MOD.wasPressed()) {
                 modEnabled = !modEnabled;

common/src/main/java/io/github/discusser/toomanyentities/config/TooManyEntitiesConfig.java

@@ -30,7 +30,6 @@ public class TooManyEntitiesConfig implements ConfigData {
     public static TooManyEntitiesConfig instance = null;
 
     public TooManyEntitiesConfig() {
-        Registries.ENTITY_TYPE.stream().forEach(entityType -> entityMaxCounts.put(entityType.getTranslationKey(), 0));
     }
 
     @Override