diff --git a/litecord/gateway/websocket.py b/litecord/gateway/websocket.py
index b758968..f596cc0 100644
--- a/litecord/gateway/websocket.py
+++ b/litecord/gateway/websocket.py
@@ -401,7 +401,7 @@ class GatewayWebsocket:
             log.exception('An exception has occoured. state={}', self.state)
             await self.ws.close(code=4000, reason=repr(err))
         finally:
-            self.ext.state_manager.remove(self)
+            self.ext.state_manager.remove(self.state)
 
             if self.state:
                 self.state.ws = None