diff --git a/host_power_off.hpp b/host_power_off.hpp
index 5b8b71f..49231a3 100644
--- a/host_power_off.hpp
+++ b/host_power_off.hpp
@@ -18,6 +18,9 @@
 
 #include <ipmid/api.h>
 
+#include <ipmid/api-types.hpp>
+#include <vector>
+
 namespace google
 {
 namespace ipmi
@@ -25,14 +28,13 @@
 
 struct HostPowerOffRequest
 {
-    uint8_t subcommand;
     // Delay in seconds.
     uint32_t delay;
 } __attribute__((packed));
 
 // Disable the fallback watchdog with given time delay and Power Off Host
-ipmi_ret_t hostPowerOff(const uint8_t* reqBuf, uint8_t* replyBuf,
-                        size_t* dataLen, const HandlerInterface* handler);
+Resp hostPowerOff(const std::vector<std::uint8_t>& data,
+                  const HandlerInterface* handler);
 
 } // namespace ipmi
 } // namespace google