2 files changed, 3 insertions, 6 deletions

diff --git a/src/ext/Bal/test/WixStdFnUnitTest/TestBootstrapperApplication.cpp b/src/ext/Bal/test/WixStdFnUnitTest/TestBootstrapperApplication.cpp
index b345ab9f..d0b38064 100644
--- a/src/ext/Bal/test/WixStdFnUnitTest/TestBootstrapperApplication.cpp
+++ b/src/ext/Bal/test/WixStdFnUnitTest/TestBootstrapperApplication.cpp
@@ -1,12 +1,11 @@
 // Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.
 
 #include "precomp.h"
-#include "BalBaseBootstrapperApplication.h"
 
-class CTestBootstrapperApplication : public CBalBaseBootstrapperApplication
+class CTestBootstrapperApplication : public CBootstrapperApplicationBase
 {
 public:
-    CTestBootstrapperApplication() : CBalBaseBootstrapperApplication()
+    CTestBootstrapperApplication() : CBootstrapperApplicationBase()
     {
     }
 };
diff --git a/src/ext/Bal/test/WixStdFnUnitTest/precomp.h b/src/ext/Bal/test/WixStdFnUnitTest/precomp.h
index 4cdd262c..23bc7a6c 100644
--- a/src/ext/Bal/test/WixStdFnUnitTest/precomp.h
+++ b/src/ext/Bal/test/WixStdFnUnitTest/precomp.h
@@ -15,10 +15,8 @@
 #include <dutil.h>
 #include <dictutil.h>
 
-#include <IBootstrapperApplication.h>
+#include <BootstrapperApplicationBase.h>
 #include <IBAFunctions.h>
-#include <balutil.h>
-#include <balretry.h>
 
 #include "TestBAFunctions.h"
 #include "TestBootstrapperApplication.h"