quick fix to websocket.py