Fix INmsBiomeAccess.getBiomeGridAt

SourceWriters · Aug 31, 2021 · 2270c78 · 2270c78
1 parent d54021c
commit 2270c78
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/...pi/src/main/java/net/sourcewriters/spigot/rwg/legacy/api/version/nms/INmsBiomeAccess.java b/...pi/src/main/java/net/sourcewriters/spigot/rwg/legacy/api/version/nms/INmsBiomeAccess.java
@@ -5,7 +5,7 @@
 import org.bukkit.block.Biome;
 
 import net.sourcewriters.spigot.rwg.legacy.api.generator.IRwgGenerator;
-import net.sourcewriters.spigot.rwg.legacy.api.util.Checks;
+import net.sourcewriters.spigot.rwg.legacy.api.generator.forward.ForwardHelper;
 import net.sourcewriters.spigot.rwg.legacy.api.util.annotation.source.NonNull;
 import net.sourcewriters.spigot.rwg.legacy.api.util.rwg.SearchBiomeGrid;
 
@@ -16,10 +16,11 @@ default SearchBiomeGrid getBiomeGrid(Chunk chunk) {
     }
 
     default SearchBiomeGrid getBiomeGridAt(World world, int x, int z) {
-        if (Checks.isRwg(world)) {
+        IRwgGenerator generator = ForwardHelper.get(world);
+        if (generator == null) {
             return null;
         }
-        return ((IRwgGenerator) world.getGenerator()).getBiomeGridAt(this, world, x, z);
+        return generator.getBiomeGridAt(this, world, x, z);
     }
 
     @NonNull