Fix disconnect callback