Fixing websocket testcase