1 files changed, 2 insertions, 2 deletions

diff --git a/src/balutil/inc/BalBaseBootstrapperApplicationProc.h b/src/balutil/inc/BalBaseBootstrapperApplicationProc.h
index 6ce8710d..d971a2eb 100644
--- a/src/balutil/inc/BalBaseBootstrapperApplicationProc.h
+++ b/src/balutil/inc/BalBaseBootstrapperApplicationProc.h
@@ -443,11 +443,11 @@ static HRESULT BalBaseBAProcOnExecuteComplete(
 
 static HRESULT BalBaseBAProcOnUnregisterBegin(
     __in IBootstrapperApplication* pBA,
-    __in BA_ONUNREGISTERBEGIN_ARGS* /*pArgs*/,
+    __in BA_ONUNREGISTERBEGIN_ARGS* pArgs,
     __inout BA_ONUNREGISTERBEGIN_RESULTS* pResults
     )
 {
-    return pBA->OnUnregisterBegin(&pResults->fCancel);
+    return pBA->OnUnregisterBegin(pArgs->fKeepRegistration, &pResults->fForceKeepRegistration);
 }
 
 static HRESULT BalBaseBAProcOnUnregisterComplete(