diff --git a/litecord/gateway/state_manager.py b/litecord/gateway/state_manager.py index 08a4114..e464cfd 100644 --- a/litecord/gateway/state_manager.py +++ b/litecord/gateway/state_manager.py @@ -247,11 +247,11 @@ class StateManager: state.ws.state = None state.ws = None - async def schedule_deletion(self, state: GatewayState): + def schedule_deletion(self, state: GatewayState): task = app.loop.create_task(self._future_cleanup(state)) self.tasks[state.session_id] = task - async def unschedule_deletion(self, state: GatewayState): + def unschedule_deletion(self, state: GatewayState): try: task = self.tasks.pop(state.session_id) except KeyError: diff --git a/litecord/gateway/websocket.py b/litecord/gateway/websocket.py index 9069102..e6b9161 100644 --- a/litecord/gateway/websocket.py +++ b/litecord/gateway/websocket.py @@ -836,7 +836,7 @@ class GatewayWebsocket: return await self.invalidate_session(False) # relink this connection - await self.app.state_manager.unschedule_deletion(state) + self.app.state_manager.unschedule_deletion(state) self.state = state state.ws = self