From d3139559f1492c8178e05826592fd2b2ba0920c7 Mon Sep 17 00:00:00 2001
From: Luke Bonaccorsi <luke@lukeb.co.uk>
Date: Tue, 8 Aug 2023 12:19:47 +0100
Subject: [PATCH] Fix loop and incorrect variable name

---
 custom_components/robovac/robovac.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/custom_components/robovac/robovac.py b/custom_components/robovac/robovac.py
index 6fbdc86..02c65c6 100644
--- a/custom_components/robovac/robovac.py
+++ b/custom_components/robovac/robovac.py
@@ -103,7 +103,7 @@ class RoboVac(TuyaDevice):
             | VacuumEntityFeature.STOP
         )
 
-        if self.modelCode in HAS_MAP_FEATURE:
+        if self.model_code in HAS_MAP_FEATURE:
             supportedFeatures |= VacuumEntityFeature.MAP
 
         return supportedFeatures
@@ -112,9 +112,9 @@ class RoboVac(TuyaDevice):
         return ROBOVAC_SERIES_FEATURES[self.getRoboVacSeries()]
 
     def getRoboVacSeries(self):
-        for series in ROBOVAC_SERIES:
-            if self.model_code in series.value:
-                return series.name
+        for series, models in ROBOVAC_SERIES:
+            if self.model_code in models:
+                return series
 
 
     def getFanSpeeds(self):