Now catching the correct exception type for when a websocket connection gets unexpectedly broken.