Fix up some websocket issues.