Fix websocket handler