Try to fix datagen hang (again)

KubeJS-Mods · May 20, 2024 · f0f9ba4 · f0f9ba4
1 parent 66df980
commit f0f9ba4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main/java/dev/latvian/mods/kubejs/helpers/MiscHelper.java b/src/main/java/dev/latvian/mods/kubejs/helpers/MiscHelper.java
@@ -9,6 +9,7 @@
 import net.minecraft.world.item.ItemStack;
 import net.neoforged.fml.ModList;
 import net.neoforged.fml.ModLoader;
+import net.neoforged.fml.loading.FMLLoader;
 import net.neoforged.fml.loading.moddiscovery.ModInfo;
 import net.neoforged.neoforge.network.IContainerFactory;
 
@@ -36,7 +37,7 @@ public void setModName(PlatformWrapper.ModInfo info, String name) {
 	}
 
 	public boolean isDataGen() {
-		return ModLoader.isDataGenRunning();
+		return FMLLoader.getLaunchHandler().isData();
 	}
 
 	public long ingotFluidAmount() {