Need to gracefully handle connection failure.