Set up to use mock websocket