[enh] Create restore hook