Updated tests for single_step.