Restore State wrapper