Saving stepData on Session - issues

[marko85to]

Hi, thanks for your extension, it's really nice.

I have experienced some issues in the "saveStep" function and I thought I would share it with everyone.
Basically I have noticed that the data stored in the session was not always updating when submitting the forms.
I've had to unset the variable in the session before updating the data, as you can see below:

WizardBehavior.php

protected function saveStep($step, $data)
    {
        if (!isset($this->_session[$this->_stepDataKey][$step])) {
            $this->_session[$this->_stepDataKey][$step] = new \ArrayObject;
        }
        unset($this->_session[$this->_stepDataKey][$step]);
        $this->_session[$this->_stepDataKey][$step][] = $data;
    }

After this update things seems working fine.