Fixing wrong callback