diff --git a/lua/gmod_integration/server/sv__websocket.lua b/lua/gmod_integration/server/sv__websocket.lua
index 9c6bec8..3402d77 100644
--- a/lua/gmod_integration/server/sv__websocket.lua
+++ b/lua/gmod_integration/server/sv__websocket.lua
@@ -48,24 +48,16 @@ function gmInte.setupWebSocket()
         gmInte.logError("WebSocket Error: " .. txt, true)
     end
 
-    function reconnect()
-        gmInte.log("WebSocket is not connected, trying to reconnect", true)
-        timer.Remove("gmInte:WebSocket:CheckConnection")
-        gmInte.setupWebSocket()
-    end
-
     timer.Create("gmInte:WebSocket:CheckConnection", 4, 0, function()
         if !socket:isConnected() then
             nbOfTry = nbOfTry + 1
             if nbOfTry > 10 && nbOfTry % 40 != 0 then return end
-            reconnect()
+            gmInte.log("WebSocket is not connected, trying to reconnect", true)
+            timer.Remove("gmInte:WebSocket:CheckConnection")
+            gmInte.setupWebSocket()
         end
     end)
-
-    hook.Add("GmodIntegration:Websocket:Restart", "gmInte:WebSocket:Restart", function()
-        socket:close()
-        reconnect()
-    end)
 end
 
+hook.Add("GmodIntegration:Websocket:Restart", "gmInte:WebSocket:Restart", function() gmInte.setupWebSocket() end)
 hook.Add("InitPostEntity", "gmInte:ServerReady:WebSocket", function() timer.Simple(1, function() gmInte.setupWebSocket() end) end)
\ No newline at end of file