Jaredhoyt cakephp-wizard

I am using the Wizard to submit an application for events. But for our use, we need to allow users to go back and edit the events they are done. I've tried using $this->Wizard->restore($dataArray), which properly loads the data from the models into the Wizard session. But then the Wizard never knows which step to go to.

My first thought was the manually set the step to step 1. When I tried this, the first step has all the data pre-loaded just like I want. But when I hit next, it doesn't know how to get to the next step because it sees that there is data in every step so it assumes that the Wizard is done and just infinitely redirects to /register/wizard.

Is there some sample code for properly restoring previously saved data and then editing it accordingly?

Thanks!