Update task done callback