Minor fix in Workflow class