Restore fixes