1 files changed, 11 insertions, 6 deletions
diff --git a/src/main/java/moe/nea/firmament/mixins/devenv/IdentifyStopperPatch.java b/src/main/java/moe/nea/firmament/mixins/devenv/IdentifyStopperPatch.java
index fac0688..cc04493 100644
--- a/src/main/java/moe/nea/firmament/mixins/devenv/IdentifyStopperPatch.java
+++ b/src/main/java/moe/nea/firmament/mixins/devenv/IdentifyStopperPatch.java
@@ -1,16 +1,21 @@
 
 package moe.nea.firmament.mixins.devenv;
 
-import net.minecraft.client.MinecraftClient;
+import net.minecraft.client.Minecraft;
 import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Shadow;
 import org.spongepowered.asm.mixin.injection.At;
 import org.spongepowered.asm.mixin.injection.Inject;
 import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
 
-@Mixin(MinecraftClient.class)
+@Mixin(Minecraft.class)
 public class IdentifyStopperPatch {
-    @Inject(method = "scheduleStop", at = @At("HEAD"))
-    private void onStop(CallbackInfo ci) {
-        Thread.dumpStack();
-    }
+	@Shadow
+	private volatile boolean running;
+
+	@Inject(method = "stop", at = @At("HEAD"))
+	private void onStop(CallbackInfo ci) {
+		if (this.running)
+			Thread.dumpStack();
+	}
 }