fix restore bug