fix restore success message not displaying