diff --git a/fw-update/device_updater.cpp b/fw-update/device_updater.cpp
index c8605f4..63e4f73 100644
--- a/fw-update/device_updater.cpp
+++ b/fw-update/device_updater.cpp
@@ -2,6 +2,9 @@
 
 #include "libpldm/firmware_update.h"
 
+#include "activation.hpp"
+#include "update_manager.hpp"
+
 #include <functional>
 
 namespace pldm
@@ -569,6 +572,7 @@
     {
         std::cout << "Component apply complete, EID=" << unsigned(eid)
                   << ", COMPONENT_VERSION=" << compVersion << "\n";
+        updateManager->updateActivationProgress();
     }
     else
     {
@@ -664,6 +668,8 @@
                   << "\n";
         return;
     }
+
+    updateManager->updateDeviceCompletion(eid, true);
 }
 
 } // namespace fw_update