pterodactyl/websocket.py

@@ -29,11 +29,8 @@ async def establish_websocket_connection(coginstance: Pterodactyl) -> None:
         logger.debug("Authentication message sent")
 
         coginstance.websocket = websocket
-
-        await loop(coginstance, websocket)
-
-async def loop(coginstance: Pterodactyl, websocket: websockets.WebSocketClientProtocol) -> None:
         current_status = ''
+
         while True: # pylint: disable=too-many-nested-blocks
             message = await websocket.recv()
             if json.loads(message)['event'] in ('token expiring', 'token expired'):