Fix protocol sorting by family (#12)

shbatm · Feb 25, 2023 · ecd0c35 · ecd0c35
1 parent 25d93a1
commit ecd0c35
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/pyisyox/nodes/__init__.py b/pyisyox/nodes/__init__.py
@@ -131,10 +131,8 @@ def parse_node_entity(self, feature: dict[str, Any]) -> None:
                     and family[TAG_ADDRESS] == NodeFamily.NODESERVER
                 ):
                     feature["node_server"] = family.get("instance", "")
-                    feature["protocol"] = self.get_protocol_from_family(
-                        feature.get(TAG_FAMILY)
-                    )
-
+                feature["protocol"] = self.get_protocol_from_family(family) 
+
             entity = Node(self, address, name, NodeDetail(**feature))
             self.add_or_update_entity(address, name, entity)
         except (TypeError, KeyError, ValueError) as exc: