Match Collapse State

New steps match the collapse state of the last step in the current model