Fix docs: State save callbacks refered to the old style state object

* Thanks to `liedekef` for pointing this error out