Fix a restore bug