diff --git a/InfrastructureAgent/windows/app-amd64-template.wxs b/InfrastructureAgent/windows/app-amd64-template.wxs
index 2bbd6a6665..8778e3abd7 100644
--- a/InfrastructureAgent/windows/app-amd64-template.wxs
+++ b/InfrastructureAgent/windows/app-amd64-template.wxs
@@ -56,5 +56,13 @@
         <Feature Id='Complete' Level='1'>
             <ComponentRef Id='MainExecutable' />
         </Feature>
+
+        <!-- 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 --> 
+        <InstallExecuteSequence> <Custom Action="ShowThankYou" After="InstallFinalize">NOT Installed</Custom> </InstallExecuteSequence>
+
     </Product>
 </Wix>
\ No newline at end of file
diff --git a/InfrastructureAgent/windows/app-arm64-template.wxs b/InfrastructureAgent/windows/app-arm64-template.wxs
index d4c28f98d5..f5d32b11f8 100644
--- a/InfrastructureAgent/windows/app-arm64-template.wxs
+++ b/InfrastructureAgent/windows/app-arm64-template.wxs
@@ -56,5 +56,13 @@
         <Feature Id='Complete' Level='1'>
             <ComponentRef Id='MainExecutable' />
         </Feature>
+
+        <!-- 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 --> 
+        <InstallExecuteSequence> <Custom Action="ShowThankYou" After="InstallFinalize">NOT Installed</Custom> </InstallExecuteSequence>
+        
     </Product>
 </Wix>
\ No newline at end of file