Fix spec error on infra_workload API