diff --git a/scripts/package_desktop.py b/scripts/package_desktop.py
index a2eeac1..737be6b 100644
--- a/scripts/package_desktop.py
+++ b/scripts/package_desktop.py
@@ -306,7 +306,7 @@ def make_msi():
       "DelFolder CUSTOM_PATH"
     ]
   else:
-    replace_in_file('DesktopEditors.aip','(<ROW Property="UpgradeCode" Value=")(.*)("/>)', r'\1%s\3' % (desktop_upgrade_code))
+    utils.replace_in_file('DesktopEditors.aip','(<ROW Property="UpgradeCode" Value=")(.*)("/>)', r'\1%s\3' % (branding.desktop_upgrade_code))
     aic_content += [
       "AddUpgradeCode '{47EEF706-B0E4-4C43-944B-E5F914B92B79}' \
                       -min_ver 7.1.1 -include_min_ver \