Add back in step (it's used for tests)