how should we hook into the websocket closing?