1 files changed, 8 insertions, 0 deletions

diff --git a/src/libs/dutil/WixToolset.DUtil/wiutil.cpp b/src/libs/dutil/WixToolset.DUtil/wiutil.cpp
index 5f81cf3a..4ed0e2c5 100644
--- a/src/libs/dutil/WixToolset.DUtil/wiutil.cpp
+++ b/src/libs/dutil/WixToolset.DUtil/wiutil.cpp
@@ -1034,6 +1034,14 @@ static DWORD CheckForRestartErrorCode(
         *pRestart = WIU_RESTART_INITIATED;
         dwErrorCode = ERROR_SUCCESS;
         break;
+
+    case ERROR_FAIL_REBOOT_REQUIRED:
+        *pRestart = WIU_RESTART_REQUIRED;
+        break;
+
+    case ERROR_FAIL_REBOOT_INITIATED:
+        *pRestart = WIU_RESTART_INITIATED;
+        break;
     }
 
     return dwErrorCode;