Fixing import WorkflowException