1 files changed, 5 insertions, 3 deletions

diff --git a/src/api/burn/balutil/inc/BalBaseBootstrapperApplication.h b/src/api/burn/balutil/inc/BalBaseBootstrapperApplication.h
index bf21c4a5..393987ba 100644
--- a/src/api/burn/balutil/inc/BalBaseBootstrapperApplication.h
+++ b/src/api/burn/balutil/inc/BalBaseBootstrapperApplication.h
@@ -435,7 +435,9 @@ public: // IBootstrapperApplication
     }
 
     virtual STDMETHODIMP OnRegisterBegin(
-        __inout BOOL* pfCancel
+        __in BOOTSTRAPPER_REGISTRATION_TYPE /*recommendedRegistrationType*/,
+        __inout BOOL* pfCancel,
+        __inout BOOTSTRAPPER_REGISTRATION_TYPE* /*pRegistrationType*/
         )
     {
         *pfCancel |= CheckCanceled();
@@ -769,8 +771,8 @@ public: // IBootstrapperApplication
     }
 
     virtual STDMETHODIMP OnUnregisterBegin(
-        __in BOOL /*fKeepRegistration*/,
-        __inout BOOL* /*pfForceKeepRegistration*/
+        __in BOOTSTRAPPER_REGISTRATION_TYPE /*recommendedRegistrationType*/,
+        __inout BOOTSTRAPPER_REGISTRATION_TYPE* /*pRegistrationType*/
         )
     {
         return S_OK;