Fixed some websocket tests

#2637