diff --git a/custom_components/robovac/robovac.py b/custom_components/robovac/robovac.py
index 6fbdc86..b5b470d 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