diff --git a/setup/setup.nsi b/setup/setup.nsi
index a92d01b..0e6dcac 100644
--- a/setup/setup.nsi
+++ b/setup/setup.nsi
@@ -419,6 +419,8 @@ ${MementoSection} "Allow OS Upgrade" ALLOWOSUPGRADE
 	WriteRegDword HKLM "${REGPATH_WUPOLICY}\OSUpgrade" "DisableOSUpgrade" 0
 	WriteRegDword HKLM "${REGPATH_WU}\OSUpgrade" "AllowOSUpgrade" 1
 	WriteRegDword HKLM "${REGPATH_WU}\OSUpgrade" "DisableOSUpgrade" 0
+	WriteRegDword HKLM "${REGPATH_WU}\OSUpgrade" "OSUpgradeInteractive" 1
+	WriteRegDword HKLM "${REGPATH_WU}\OSUpgrade" "OSUpgradeRunOnceCount" 1
 	Call RebootIfRequired
 ${MementoSectionEnd}