onDisconnect fix