diff --git a/ubi/activation_ubi.cpp b/ubi/activation_ubi.cpp
index 946a741..205c393 100644
--- a/ubi/activation_ubi.cpp
+++ b/ubi/activation_ubi.cpp
@@ -129,7 +129,7 @@
     activationProgress->progress(10);
 }
 
-void ActivationUbi::unitStateChange(sdbusplus::message::message& msg)
+void ActivationUbi::unitStateChange(sdbusplus::message_t& msg)
 {
     uint32_t newStateID{};
     sdbusplus::message::object_path newStateObjPath;