Fixed some websocket stuff