From cdeb107d243e0fb1ae238d4f92c09e29c1b41b46 Mon Sep 17 00:00:00 2001
From: Aso <53526710+AsoDesu@users.noreply.github.com>
Date: Thu, 6 Jul 2023 18:28:45 +0100
Subject: [PATCH] Oops

---
 .../mixins/discord/ConnectionMixin.java         | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/src/main/java/net/asodev/islandutils/mixins/discord/ConnectionMixin.java b/src/main/java/net/asodev/islandutils/mixins/discord/ConnectionMixin.java
index af4612e..bc39b98 100644
--- a/src/main/java/net/asodev/islandutils/mixins/discord/ConnectionMixin.java
+++ b/src/main/java/net/asodev/islandutils/mixins/discord/ConnectionMixin.java
@@ -21,13 +21,16 @@ public abstract class ConnectionMixin {
 
     @Inject(method = "disconnect", at = @At("HEAD"))
     private void disconnect(Component component, CallbackInfo ci) {
-        if (getRemoteAddress() == null) return;
-        String hostName = ((InetSocketAddress) getRemoteAddress()).getHostName();
-        if (hostName == null) return;
-        if (hostName.contains("mccisland.net")) {
-            DiscordPresenceUpdator.started = null;
-            DiscordPresenceUpdator.clear();
-            MccIslandState.setGame(GAME.HUB);
+        SocketAddress remoteAddress = getRemoteAddress();
+        if (remoteAddress == null) return;
+        if (remoteAddress instanceof InetSocketAddress socketAddress) {
+            String hostName = socketAddress.getHostName();
+            if (hostName == null) return;
+            if (hostName.contains("mccisland.net")) {
+                DiscordPresenceUpdator.started = null;
+                DiscordPresenceUpdator.clear();
+                MccIslandState.setGame(GAME.HUB);
+            }
         }
     }