Update websocket_endpoint.go