Add missing regen code