diff --git a/custom_components/robovac/vacuums/T2267.py b/custom_components/robovac/vacuums/T2267.py
index f4e8fc7..1f20fe4 100644
--- a/custom_components/robovac/vacuums/T2267.py
+++ b/custom_components/robovac/vacuums/T2267.py
@@ -2,7 +2,7 @@
 from .base import RoboVacEntityFeature, RobovacCommand
 
 
-class T2277:
+class T2267:
     homeassistant_features = (
         VacuumEntityFeature.BATTERY
 #        | VacuumEntityFeature.CLEAN_SPOT
diff --git a/custom_components/robovac/vacuums/T2278.py b/custom_components/robovac/vacuums/T2278.py
index f4e8fc7..c4b1ce0 100644
--- a/custom_components/robovac/vacuums/T2278.py
+++ b/custom_components/robovac/vacuums/T2278.py
@@ -2,7 +2,7 @@
 from .base import RoboVacEntityFeature, RobovacCommand
 
 
-class T2277:
+class T2278:
     homeassistant_features = (
         VacuumEntityFeature.BATTERY
 #        | VacuumEntityFeature.CLEAN_SPOT
diff --git a/custom_components/robovac/vacuums/__init__.py b/custom_components/robovac/vacuums/__init__.py
index 880b536..2706f3c 100644
--- a/custom_components/robovac/vacuums/__init__.py
+++ b/custom_components/robovac/vacuums/__init__.py
@@ -29,6 +29,8 @@
 from .T2193 import T2193
 from .T2194 import T2194
 from .T2267 import T2267
+from .T2277 import T2277
+from .T2278 import T2278
 from .T2261 import T2261
 from .T2262 import T2262
 from .T2320 import T2320
@@ -66,6 +68,8 @@
     "T2193": T2193,
     "T2194": T2194,
     "T2267": T2267,
+    "T2277": T2277,
+    "T2278": T2278,
     "T2261": T2261,
     "T2262": T2262,
     "T2320": T2320