Update to reconnect logic