Updated websocket plugin