diff --git a/inc/button_handler.hpp b/inc/button_handler.hpp
index 819a433..1f74f18 100644
--- a/inc/button_handler.hpp
+++ b/inc/button_handler.hpp
@@ -91,7 +91,7 @@
      * @param[in] msg - sdbusplus message from signal
      */
 
-    void debugHostSelectorReleased(sdbusplus::message::message& msg);
+    void debugHostSelectorReleased(sdbusplus::message_t& msg);
 
     /**
      * @brief Checks if system is powered on
diff --git a/inc/debugHostSelector_button.hpp b/inc/debugHostSelector_button.hpp
index bdd9dfa..fa13d02 100644
--- a/inc/debugHostSelector_button.hpp
+++ b/inc/debugHostSelector_button.hpp
@@ -24,8 +24,8 @@
 
 {
   public:
-    DebugHostSelector(sdbusplus::bus::bus& bus, const char* path,
-                      EventPtr& event, buttonConfig& buttonCfg) :
+    DebugHostSelector(sdbusplus::bus_t& bus, const char* path, EventPtr& event,
+                      buttonConfig& buttonCfg) :
         sdbusplus::server::object_t<
             sdbusplus::xyz::openbmc_project::Chassis::Buttons::server::Button>(
             bus, path, action::defer_emit),