fix disconnect  bug