fixed minor disconnect bug