Change DisconnectAsync to use extension instead

diff --git a/src/Impostor.Server/Net/ClientBase.cs b/src/Impostor.Server/Net/ClientBase.cs
index 74224fcc4475bbe8d18d1b88d161f31eea110dee..15dfe91982b65078a5af0f55c59cac7944876705 100644 (file)
--- a/src/Impostor.Server/Net/ClientBase.cs
+++ b/src/Impostor.Server/Net/ClientBase.cs
@@ -5,8 +5,6 @@ using Impostor.Api;
 using Impostor.Api.Innersloth;
 using Impostor.Api.Net;
 using Impostor.Api.Net.Messages;
-using Impostor.Api.Net.Messages.S2C;
-using Impostor.Hazel;
 using Impostor.Server.Net.State;
 
 namespace Impostor.Server.Net
@@ -55,15 +53,7 @@ namespace Impostor.Server.Net
 
         public async ValueTask DisconnectAsync(DisconnectReason reason, string? message = null)
         {
-            if (!Connection.IsConnected)
-            {
-                return;
-            }
-
-            using var writer = MessageWriter.Get();
-            MessageDisconnect.Serialize(writer, true, reason, message);
-
-            await Connection.DisconnectAsync(message ?? reason.ToString(), writer);
+            await Connection.CustomDisconnectAsync(message ?? reason.ToString());
         }
     }
 }