Minor web socket cleanup