'state' is no longer allowed to be passed in session. Fix specs to match.

All green now! @homakov, please have a look.