Reload on connection lost refactor