Correct function return

Author: bouwew

plugwise/__init__.py

@@ -193,18 +193,20 @@ async def _smile_detect(
         self._store_smile_data(model)
         self._process_for_thermostat(result)
 
-    async def _collect_smile_data(self, dsmrmain: etree.Element, result: etree.Element) -> str:
+    async def _collect_smile_data(
+        self, dsmrmain: etree.Element, result: etree.Element
+    ) -> str:
         """Collect smile/gateway data."""
         model: str = "Unknown"
         if (gateway := result.find("./gateway")) is not None:
             self.smile.version = parse(gateway.find("firmware_version").text)
             self.smile.hw_version = gateway.find("hardware_version").text
             self.smile.hostname = gateway.find("hostname").text
             self.smile.mac_address = gateway.find("mac_address").text
-            if (vendor_model := gateway.find("vendor_model")) is None:
-                return  # pragma: no cover
+            if (vendor_model := gateway.find("vendor_model")) is not None:
+                model = vendor_model.text
 
-            model = vendor_model.text
+            # Check for Anna P1 function
             elec_point_meters = result.findall(
                 "./location/logs/point_log/electricity_point_meter"
             )