4 files changed, 5 insertions, 8 deletions

diff --git a/src/test/burn/TestData/BAFunctionsTests/BafRelatedBundleVariableTesting/precomp.h b/src/test/burn/TestData/BAFunctionsTests/BafRelatedBundleVariableTesting/precomp.h
index 2e14786a..95a015ab 100644
--- a/src/test/burn/TestData/BAFunctionsTests/BafRelatedBundleVariableTesting/precomp.h
+++ b/src/test/burn/TestData/BAFunctionsTests/BafRelatedBundleVariableTesting/precomp.h
@@ -30,8 +30,7 @@
 #include "regutil.h"
 #include "xmlutil.h"
 
-#include "BalBaseBootstrapperApplication.h"
-#include "balutil.h"
+#include <BootstrapperApplicationBase.h>
 
 #include "BAFunctions.h"
 #include "IBAFunctions.h"
diff --git a/src/test/burn/TestData/Manual/BafThmutilTesting/precomp.h b/src/test/burn/TestData/Manual/BafThmutilTesting/precomp.h
index 2e14786a..95a015ab 100644
--- a/src/test/burn/TestData/Manual/BafThmutilTesting/precomp.h
+++ b/src/test/burn/TestData/Manual/BafThmutilTesting/precomp.h
@@ -30,8 +30,7 @@
 #include "regutil.h"
 #include "xmlutil.h"
 
-#include "BalBaseBootstrapperApplication.h"
-#include "balutil.h"
+#include <BootstrapperApplicationBase.h>
 
 #include "BAFunctions.h"
 #include "IBAFunctions.h"
diff --git a/src/test/burn/TestData/PrereqBaTests/PrereqBaf/PrereqBaf.cpp b/src/test/burn/TestData/PrereqBaTests/PrereqBaf/PrereqBaf.cpp
index 5e26faac..1eaebe63 100644
--- a/src/test/burn/TestData/PrereqBaTests/PrereqBaf/PrereqBaf.cpp
+++ b/src/test/burn/TestData/PrereqBaTests/PrereqBaf/PrereqBaf.cpp
@@ -9,7 +9,7 @@ class CPrereqBaf : public CBalBaseBAFunctions
 public: // IBAFunctions
 
 public: //IBootstrapperApplication
-    STDMETHODIMP OnCreate(
+    virtual STDMETHODIMP OnCreate(
         __in IBootstrapperEngine* pEngine,
         __in BOOTSTRAPPER_COMMAND* pCommand
     )
@@ -17,7 +17,7 @@ public: //IBootstrapperApplication
         HRESULT hr = S_OK;
 
         hr = __super::OnCreate(pEngine, pCommand);
-        ExitOnFailure(hr, "CBalBaseBootstrapperApplication initialization failed.");
+        ExitOnFailure(hr, "CPrereqBaf initialization failed.");
 
         hr = StrAllocString(&m_sczBARuntimeDirectory, pCommand->wzBootstrapperWorkingFolder, 0);
         ExitOnFailure(hr, "Failed to copy working folder");
diff --git a/src/test/burn/TestData/PrereqBaTests/PrereqBaf/precomp.h b/src/test/burn/TestData/PrereqBaTests/PrereqBaf/precomp.h
index 8320bdd8..f39f9de7 100644
--- a/src/test/burn/TestData/PrereqBaTests/PrereqBaf/precomp.h
+++ b/src/test/burn/TestData/PrereqBaTests/PrereqBaf/precomp.h
@@ -18,8 +18,7 @@
 #include "pathutil.h"
 #include "strutil.h"
 
-#include "BalBaseBootstrapperApplication.h"
-#include "balutil.h"
+#include <BootstrapperApplicationBase.h>
 
 #include "BAFunctions.h"
 #include "IBAFunctions.h"