Poprawienie funkcji disconnect