fix re-connection callback issue