Updated infer output args