Fixed test_task_arg_gen