Add AppWaitForSingleObject/MultipleObjects, ThreadWaitForCompletion.

2 files changed, 2 insertions, 2 deletions

diff --git a/src/burn/test/BurnUnitTest/ElevationTest.cpp b/src/burn/test/BurnUnitTest/ElevationTest.cpp
index 97d76b7d..ce34b4b1 100644
--- a/src/burn/test/BurnUnitTest/ElevationTest.cpp
+++ b/src/burn/test/BurnUnitTest/ElevationTest.cpp
@@ -54,7 +54,6 @@ namespace Bootstrapper
             HRESULT hr = S_OK;
             BURN_ENGINE_STATE engineState = { };
             BURN_PIPE_CONNECTION* pConnection = &engineState.companionConnection;
-            HANDLE hEvent = NULL;
             DWORD dwResult = S_OK;
 
             engineState.sczBundleEngineWorkingPath = L"tests\\ignore\\this\\path\\to\\burn.exe";
@@ -62,6 +61,7 @@ namespace Bootstrapper
             try
             {
                 ShelFunctionOverride(ElevateTest_ShellExecuteExW);
+                CoreFunctionOverride(ThrdWaitForCompletion);
 
                 PipeConnectionInitialize(pConnection);
 
@@ -87,7 +87,6 @@ namespace Bootstrapper
             finally
             {
                 PipeConnectionUninitialize(pConnection);
-                ReleaseHandle(hEvent);
             }
         }
     };
diff --git a/src/burn/test/BurnUnitTest/precomp.h b/src/burn/test/BurnUnitTest/precomp.h
index 92c8c4c0..69c62cb2 100644
--- a/src/burn/test/BurnUnitTest/precomp.h
+++ b/src/burn/test/BurnUnitTest/precomp.h
@@ -28,6 +28,7 @@
 #include <resrutil.h>
 #include <shelutil.h>
 #include <strutil.h>
+#include <thrdutil.h>
 #include <wiutil.h>
 #include <xmlutil.h>
 #include <dictutil.h>