fix unit tests for new version of stagerunner