Use proper websocket module