diff --git a/arclight-common/src/main/java/io/izzel/arclight/common/mod/util/DistValidate.java b/arclight-common/src/main/java/io/izzel/arclight/common/mod/util/DistValidate.java
index e4edf324..97536567 100644
--- a/arclight-common/src/main/java/io/izzel/arclight/common/mod/util/DistValidate.java
+++ b/arclight-common/src/main/java/io/izzel/arclight/common/mod/util/DistValidate.java
@@ -32,9 +32,8 @@ public static boolean isValid(BlockGetter getter) {
     }
 
     private static boolean isLogicWorld(LevelAccessor level) {
-        var cl = level.getClass();
-        return cl == ServerLevel.class || cl == WorldGenRegion.class
-            || isLogicWorld(cl);
+        return level instanceof ServerLevel || level instanceof WorldGenRegion
+            || isLogicWorld(level.getClass());
     }
 
     private static final Map<Class<?>, Boolean> SEEN_CLASSES = new ConcurrentHashMap<>();