diff --git a/host_state_manager.hpp b/host_state_manager.hpp
index 597efa5..4678746 100644
--- a/host_state_manager.hpp
+++ b/host_state_manager.hpp
@@ -1,5 +1,6 @@
 #pragma once
 
+#include <string>
 #include <sdbusplus/bus.hpp>
 #include "xyz/openbmc_project/State/Host/server.hpp"
 
@@ -86,6 +87,13 @@
          */
         void executeTransition(Transition tranReq);
 
+        /**
+         * @brief Determine if auto reboot flag is set
+         *
+         * @return boolean corresponding to current auto_reboot setting
+         **/
+        bool isAutoReboot();
+
         /** @brief Callback function on systemd state changes
          *
          * Will just do a call into the appropriate object for processing