Gracefully handle server going away