From fde4f1a8bd6582572c634f34a4ac5ac28d4ecee1 Mon Sep 17 00:00:00 2001
From: Mgazul <mgazul@163.com>
Date: Fri, 26 Jan 2024 20:44:55 +0700
Subject: [PATCH] Fix dimension register

---
 src/main/java/com/mohistmc/forge/ForgeInjectBukkit.java | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/main/java/com/mohistmc/forge/ForgeInjectBukkit.java b/src/main/java/com/mohistmc/forge/ForgeInjectBukkit.java
index 98e140f08f..f65303d90a 100644
--- a/src/main/java/com/mohistmc/forge/ForgeInjectBukkit.java
+++ b/src/main/java/com/mohistmc/forge/ForgeInjectBukkit.java
@@ -216,12 +216,13 @@ public static void addEnumEnvironment(Registry<Dimension> registry) {
         int i = World.Environment.values().length;
         for (Map.Entry<RegistryKey<Dimension>, Dimension> entry : registry.entrySet()) {
             RegistryKey<Dimension> key = entry.getKey();
-            World.Environment environment1 = environment.get(key);
+            DimensionType dimensionType = entry.getValue().type();
+            World.Environment environment1 = environment.get(dimensionType);
             if (environment1 == null) {
                 String name = normalizeName(key.location().toString());
                 int id = i - 1;
                 environment1 = MohistEnumHelper.addEnum(World.Environment.class, name, new Class[]{Integer.TYPE}, new Object[]{id});
-                environment.put(entry.getValue().type(), environment1);
+                environment.put(dimensionType, environment1);
                 environment0.put(environment1, key);
                 MohistMC.LOGGER.debug("Registered forge DimensionType as environment {}", environment1);
                 i++;