1 files changed, 3 insertions, 2 deletions

diff --git a/src/balutil/inc/BalBaseBAFunctionsProc.h b/src/balutil/inc/BalBaseBAFunctionsProc.h
index 4f0906d2..1b11c300 100644
--- a/src/balutil/inc/BalBaseBAFunctionsProc.h
+++ b/src/balutil/inc/BalBaseBAFunctionsProc.h
@@ -58,12 +58,12 @@ static HRESULT WINAPI BalBaseBAFunctionsProc(
         case BA_FUNCTIONS_MESSAGE_ONDETECTRELATEDBUNDLE:
         case BA_FUNCTIONS_MESSAGE_ONDETECTPACKAGEBEGIN:
         case BA_FUNCTIONS_MESSAGE_ONDETECTRELATEDMSIPACKAGE:
-        case BA_FUNCTIONS_MESSAGE_ONDETECTTARGETMSIPACKAGE:
+        case BA_FUNCTIONS_MESSAGE_ONDETECTPATCHTARGET:
         case BA_FUNCTIONS_MESSAGE_ONDETECTMSIFEATURE:
         case BA_FUNCTIONS_MESSAGE_ONDETECTPACKAGECOMPLETE:
         case BA_FUNCTIONS_MESSAGE_ONPLANRELATEDBUNDLE:
         case BA_FUNCTIONS_MESSAGE_ONPLANPACKAGEBEGIN:
-        case BA_FUNCTIONS_MESSAGE_ONPLANTARGETMSIPACKAGE:
+        case BA_FUNCTIONS_MESSAGE_ONPLANPATCHTARGET:
         case BA_FUNCTIONS_MESSAGE_ONPLANMSIFEATURE:
         case BA_FUNCTIONS_MESSAGE_ONPLANPACKAGECOMPLETE:
         case BA_FUNCTIONS_MESSAGE_ONAPPLYBEGIN:
@@ -107,6 +107,7 @@ static HRESULT WINAPI BalBaseBAFunctionsProc(
         case BA_FUNCTIONS_MESSAGE_ONPAUSEAUTOMATICUPDATESCOMPLETE:
         case BA_FUNCTIONS_MESSAGE_ONSYSTEMRESTOREPOINTBEGIN:
         case BA_FUNCTIONS_MESSAGE_ONSYSTEMRESTOREPOINTCOMPLETE:
+        case BA_FUNCTIONS_MESSAGE_ONPLANNEDPACKAGE:
             hr = BalBaseBootstrapperApplicationProc((BOOTSTRAPPER_APPLICATION_MESSAGE)message, pvArgs, pvResults, pvContext);
             break;
         case BA_FUNCTIONS_MESSAGE_ONTHEMELOADED: