Improve WebSocketAction