diff --git a/custom_components/viomise/vacuum.py b/custom_components/viomise/vacuum.py
index 9a35e1d..8f5b346 100644
--- a/custom_components/viomise/vacuum.py
+++ b/custom_components/viomise/vacuum.py
@@ -146,8 +146,8 @@
     "filter_left",
     "mop_percentage",
     "mop_left",
-    "repeat-state",
-    "mop-route"
+    "repeat_state",
+    "mop_route"
 ]
 
 async def async_setup_platform(hass, config, async_add_entities, discovery_info=None):