diff --git a/setup/WinDirStat.wxs b/setup/WinDirStat.wxs
index 680881c7..6fe5ea3c 100644
--- a/setup/WinDirStat.wxs
+++ b/setup/WinDirStat.wxs
@@ -40,12 +40,12 @@
- "") OR ENABLE_SHORTCUT)]]>
+ "" AND DESKTOP_SHORTCUT_LOCATE <> "") OR ENABLE_SHORTCUT = 1)]]>
+ "" AND CONTEXT_MENU_LOCATE <> "") OR NOT WIX_UPGRADE_DETECTED) AND DISABLE_CONTEXTMENU <> 1]]>