Parameterize the task chain callback correctly