Implementing this correctly would be better.  I moved tests for stage 3 from `build_step` to `test_step`.