Fix connection loss