diff --git a/confluent_server/confluent/messages.py b/confluent_server/confluent/messages.py
index 0e92c815..767d9f8b 100644
--- a/confluent_server/confluent/messages.py
+++ b/confluent_server/confluent/messages.py
@@ -538,6 +538,7 @@ class InputPowerMessage(ConfluentInputMessage):
         'off',
         'reset',
         'boot',
+        'diag',
         'shutdown',
     ])
 
@@ -650,6 +651,7 @@ class PowerState(ConfluentChoiceMessage):
         'off',
         'reset',
         'boot',
+        'diag',
     ])
     keyname = 'state'