Fixed test to reflect recent step model