[enh] Restore function