Index: sources/update/UiSwUpdate.cpp
===================================================================
diff -u -r20b370a54d2737831b307a0de82aec9e06e2b772 -r7cf31f6cfef97bb564a103c263821a3b722c8a2b
--- sources/update/UiSwUpdate.cpp	(.../UiSwUpdate.cpp)	(revision 20b370a54d2737831b307a0de82aec9e06e2b772)
+++ sources/update/UiSwUpdate.cpp	(.../UiSwUpdate.cpp)	(revision 7cf31f6cfef97bb564a103c263821a3b722c8a2b)
@@ -16,6 +16,8 @@
  
 #include "UiSwUpdate.h"
 #include "IDataProvider.h"
+#include "IEnterBootLoader.h"
+
 #include <algorithm>
 
 namespace SwUpdate {
@@ -67,12 +69,13 @@
 /*!
  * \brief   Start updating.
  *
- * \param   dataProviders
+ * \param   dataProviders   Vector of Data providers.
+ * \param   pEnterBL        Class to call if need to enter bootloader.
  *
  * \return  True on successfully started.
  */
 bool UiSwUpdate::start(
-    std::vector<IDataProvider*> & dataProviders) {
+    std::vector<IDataProvider*> & dataProviders, SwUpdate::IEnterBootLoader * pEnterBL) {
 
     // Lock.
     std::lock_guard<std::mutex> lock(_mutexApi);
@@ -104,7 +107,7 @@
         // Set the provider.
         UiProtocol* pTarget = _all[target];
         aboutToStart[target] = pTarget;
-        _all[target]->setProvider(pDp);
+        _all[target]->setProvider(pDp, pEnterBL);
 
         // Ensure all non-file actions happen after the tasks.
         priorTargets.push_back(pTarget);