Fix web socket handler