Removed websocket from unit test