diff --git a/midealocal/devices/b6/message.py b/midealocal/devices/b6/message.py
index 8bbf4ea7..58898403 100644
--- a/midealocal/devices/b6/message.py
+++ b/midealocal/devices/b6/message.py
@@ -15,6 +15,8 @@
 FAN_LEVEL_RANGE_3 = 170
 MIN_FAN_LEVEL_RANGE = 100
 
+MESSAGE_PROTOCOL_VERSION = 2
+
 
 class MessageB6Base(MessageRequest):
     """B6 message base."""
@@ -46,7 +48,9 @@ def __init__(self, protocol_version: int) -> None:
         super().__init__(
             protocol_version=protocol_version,
             message_type=MessageType.query,
-            body_type=BodyType.X11 if protocol_version == 2 else BodyType.X31,
+            body_type=BodyType.X11
+            if protocol_version == MESSAGE_PROTOCOL_VERSION
+            else BodyType.X31,
         )
 
     @property