Fixed disconnect bug.