Fix web socket tests