Sorry, I know I nag on this minor point, but could you please do the `if workflow is None: return...` and put the continue_grading afterwards, outside the `if`?