More gracefully handle disconnection errors