fix restoration issue