Make sure to use latest version_to_model function

Author: bouwew

plugwise/util.py

@@ -36,13 +36,16 @@ def validate_mac(mac):
 
 def version_to_model(version):
     """Translate hardware_version to device type."""
-    model = HW_MODELS.get(version[4:10], None)
+    if version is None:
+        return None
+
+    model = HW_MODELS.get(version)
+    if model is None:
+        model = HW_MODELS.get(version[4:10])
     if model is None:
         # Try again with reversed order
-        model = HW_MODELS.get(
-            version[-2:] + version[-4:-2] + version[-6:-4],
-            None,
-        )
+        model = HW_MODELS.get(version[-2:] + version[-4:-2] + version[-6:-4])
+
     return model if model is not None else "Unknown"