From 4e2ead95f478f598a2dc3f9fccd98b79f8bae642 Mon Sep 17 00:00:00 2001
From: MrGeorgen <moinl6162@gmail.com>
Date: Wed, 27 Jan 2021 23:09:23 +0100
Subject: [PATCH] fixed disconnect screen message

---
 src/main/java/de/mrgeorgen/wolfSignal/Main.java                  | 1 +
 .../java/de/mrgeorgen/wolfSignal/mixin/MixinMinecraftClient.java | 1 +
 2 files changed, 2 insertions(+)

diff --git a/src/main/java/de/mrgeorgen/wolfSignal/Main.java b/src/main/java/de/mrgeorgen/wolfSignal/Main.java
index bab1c37..ac29fe3 100644
--- a/src/main/java/de/mrgeorgen/wolfSignal/Main.java
+++ b/src/main/java/de/mrgeorgen/wolfSignal/Main.java
@@ -29,6 +29,7 @@ public class Main implements ModInitializer {
 
 	private static int cmdSignal(CommandContext<ClientCommandSource> ctx) {
 		signal = ctx.getArgument("signal", int.class);
+		disconnectedCausedByWolfSignal = true;
 		MinecraftClient.getInstance().world.disconnect();
 		signalIndex = 4;
 		setTick();
diff --git a/src/main/java/de/mrgeorgen/wolfSignal/mixin/MixinMinecraftClient.java b/src/main/java/de/mrgeorgen/wolfSignal/mixin/MixinMinecraftClient.java
index 3ce5f0f..fdd1e88 100644
--- a/src/main/java/de/mrgeorgen/wolfSignal/mixin/MixinMinecraftClient.java
+++ b/src/main/java/de/mrgeorgen/wolfSignal/mixin/MixinMinecraftClient.java
@@ -56,6 +56,7 @@ public class MixinMinecraftClient
 					(currentScreen instanceof ConnectScreen && !(newScreen instanceof DisconnectedScreen))) // connection successful or cancelled using cancel button on connect screen
 			{
 				ticks = -1;
+				disconnectedCausedByWolfSignal = false;
 			}
 		}
 	}