Finally fixed disconnect bug

Yep yep