Fix websocket tests