diff --git a/src/main/java/de/pilz/plantmegapackfix/mixins/late/PMPWorldGeneratorMixin.java b/src/main/java/de/pilz/plantmegapackfix/mixins/late/PMPWorldGeneratorMixin.java
index dd09108..2ba5314 100644
--- a/src/main/java/de/pilz/plantmegapackfix/mixins/late/PMPWorldGeneratorMixin.java
+++ b/src/main/java/de/pilz/plantmegapackfix/mixins/late/PMPWorldGeneratorMixin.java
@@ -35,7 +35,7 @@ private void generateEnd(World world, Random random, int chunkX, int chunkZ) {}
             // Handled by original already
             return;
         }
-        if (world.provider instanceof WorldProviderSurface) {
+        if (world.provider.isSurfaceWorld()) {
             if (PlantMegaPack.settingsGeneral.worldgenOverworldRate > 0) {
                 generateSurface(world, random, chunkX * 16, chunkZ * 16);
             }