fix restore function