Patches up the new updating logic.