diff --git a/command/session_cmds.hpp b/command/session_cmds.hpp
index d5d394b..026b9d8 100644
--- a/command/session_cmds.hpp
+++ b/command/session_cmds.hpp
@@ -77,7 +77,7 @@
  */
 std::vector<uint8_t>
     setSessionPrivilegeLevel(const std::vector<uint8_t>& inPayload,
-                             const message::Handler& handler);
+                             std::shared_ptr<message::Handler>& handler);
 /**
  * @struct CloseSessionRequest
  *
@@ -115,6 +115,6 @@
  * @return Response data for the command
  */
 std::vector<uint8_t> closeSession(const std::vector<uint8_t>& inPayload,
-                                  const message::Handler& handler);
+                                  std::shared_ptr<message::Handler>& handler);
 
 } // namespace command