Removed the import for websocket