Update websocket.go