From dd0c080ff8c993d6d9de395f4ead51e1b1320376 Mon Sep 17 00:00:00 2001
From: Paul <pbuckley4192@gmail.com>
Date: Tue, 8 Aug 2023 10:00:37 +0100
Subject: [PATCH] Fix for bad array loop

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

diff --git a/custom_components/robovac/robovac.py b/custom_components/robovac/robovac.py
index b5b470d..47edc9e 100644
--- a/custom_components/robovac/robovac.py
+++ b/custom_components/robovac/robovac.py
@@ -19,7 +19,7 @@ class RoboVacEntityFeature(IntEnum):
     BOOST_IQ = 1024
 
 
-HAS_MAP_FEATURE = ["T2261", "T2262"]
+HAS_MAP_FEATURE = ["T2253","T2261", "T2262"]
 
 ROBOVAC_SERIES = {
     "C": [
@@ -113,8 +113,8 @@ class RoboVac(TuyaDevice):
 
     def getRoboVacSeries(self):
         for series in ROBOVAC_SERIES:
-            if self.model_code in series.value:
-                return series.name
+            if self.model_code in ROBOVAC_SERIES[series]:
+                return series
 
 
     def getFanSpeeds(self):