Fixed to use # of steps rather than # of states.