Set the callback when notifyOnCancel is called

We weren't actually setting the callback field in notifyOnCancel