Tweak restore process