Fix websocket test