diff --git a/host_state_manager.cpp b/host_state_manager.cpp
index cf8105f..bf5e79a 100644
--- a/host_state_manager.cpp
+++ b/host_state_manager.cpp
@@ -26,6 +26,7 @@
 #include <fstream>
 #include <iostream>
 #include <map>
+#include <set>
 #include <string>
 
 // Register class version with Cereal
@@ -84,6 +85,14 @@
     return;
 }
 
+void Host::setupSupportedTransitions()
+{
+    std::set<Transition> supportedTransitions = {
+        Transition::On, Transition::Off, Transition::Reboot,
+        Transition::GracefulWarmReboot, Transition::ForceWarmReboot};
+    server::Host::allowedHostTransitions(supportedTransitions);
+}
+
 void Host::createSystemdTargetMaps()
 {
     stateTargetTable = {