session restore fixes