fix restore problem