fix state manager removal

2018-07-21 09:35:55 -07:00 · 2018-07-21 09:35:55 -07:00 · bc9d1e8908
parent 681e35db5a
commit bc9d1e8908
1 changed files with 1 additions and 1 deletions
--- 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