Fixed disconnect bug