diff --git a/InfrastructureAgent/windows/app-amd64-template.wxs b/InfrastructureAgent/windows/app-amd64-template.wxs
index 16b65cd5cb..e943d042f7 100644
--- a/InfrastructureAgent/windows/app-amd64-template.wxs
+++ b/InfrastructureAgent/windows/app-amd64-template.wxs
@@ -60,8 +60,6 @@
     <CustomAction Id='SetEnvironment' Directory='TARGETDIR' ExeCommand='cmd.exe /c setx PATH [INSTALLDIR]' Return='check' />
 
 
-        <!-- Include the WixUIExtension for standard dialogs --> 
-        <UIRef Id="WixUI_InstallDir" /> 
         <!-- Add a custom action to show a thank you message --> 
         <CustomAction Id="ShowThankYou" Return="ignore" Execute="immediate" Script="vbscript"> <![CDATA[ MsgBox "Thank you for installing OneUptimevInfrastructure Agent!", vbInformation, "Installation Complete" ]]> </CustomAction> 
         <!-- Schedule the custom action to run at the end of the installation --> 
diff --git a/InfrastructureAgent/windows/app-arm64-template.wxs b/InfrastructureAgent/windows/app-arm64-template.wxs
index 1f6cc1f4df..a22be0ecb1 100644
--- a/InfrastructureAgent/windows/app-arm64-template.wxs
+++ b/InfrastructureAgent/windows/app-arm64-template.wxs
@@ -61,8 +61,6 @@
     <CustomAction Id='SetEnvironment' Directory='TARGETDIR' ExeCommand='cmd.exe /c setx PATH [INSTALLDIR]' Return='check' />
 
 
-        <!-- Include the WixUIExtension for standard dialogs --> 
-        <UIRef Id="WixUI_InstallDir" /> 
         <!-- Add a custom action to show a thank you message --> 
         <CustomAction Id="ShowThankYou" Return="ignore" Execute="immediate" Script="vbscript"> <![CDATA[ MsgBox "Thank you for installing OneUptimevInfrastructure Agent!", vbInformation, "Installation Complete" ]]> </CustomAction> 
         <!-- Schedule the custom action to run at the end of the installation -->