Fix another transition when sampling workflow is on

The refactor to utils/*.py left these hanging