diff --git a/custom_components/dyson_local/vacuum.py b/custom_components/dyson_local/vacuum.py
index 2ede64c..879f65c 100644
--- a/custom_components/dyson_local/vacuum.py
+++ b/custom_components/dyson_local/vacuum.py
@@ -131,7 +131,7 @@
     VacuumVisNavPowerMode.BOOST: "Boost",
 }
 VIS_NAV_POWER_MODE_STR_TO_ENUM = {
-    value: key for key, value in HEURIST_POWER_MODE_ENUM_TO_STR.items()
+    value: key for key, value in VIS_NAV_POWER_MODE_ENUM_TO_STR.items()
 }
 
 ATTR_POSITION = "position"