Fixed lost connection problem