Fixing outdated callback reference