diff --git a/rmp220_middleware/rmp220_middleware.py b/rmp220_middleware/rmp220_middleware.py
index 33f5b61..d3a6643 100644
--- a/rmp220_middleware/rmp220_middleware.py
+++ b/rmp220_middleware/rmp220_middleware.py
@@ -64,14 +64,17 @@ class StateMachineNode(Node):
         # Need to figure out the type
         if msg.chassis_mode == 0:
             self.chassis_mode = State.DISABLED
+            self.state = State.DISABLED
         if msg.chassis_mode == 1:  # Assuming 1 represents enabled and 0 represents disabled
             self.chassis_mode = State.ENABLED
+            self.state = State.ENABLED
         if msg.chassis_mode == 2:
             self.chassis_mode = State.PASSIVE
+            self.state = State.PASSIVE
         if msg.chassis_mode == 3:
             self.chassis_mode = State.STOPPED
+            self.state = State.STOPPED
         
-
     def get_chassis_mode(self):
         return self.chassis_mode