Fix websocket impl