Fix - clean up websocket when user disconnects