Remove set_task_expansion calls