Add rescue from connection failure wrapper