Fixed disconnect issue.