Update task_utils.py