Change @task assignment to fix buggy code