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