diff --git a/packages/vacuum/vacuum.yaml b/packages/vacuum/vacuum.yaml
index 05b5fba..deaf3e4 100644
--- a/packages/vacuum/vacuum.yaml
+++ b/packages/vacuum/vacuum.yaml
@@ -81,25 +81,25 @@ input_boolean:
 
 homeassistant:
   customize:
-    furbot_segment_bathroom:
+    input_boolean.furbot_segment_bathroom:
       segment_id: '1'
-    furbot_segment_bedroom:
+    input_boolean.furbot_segment_bedroom:
       segment_id: '2'
-    furbot_segment_guest_room:
+    input_boolean.furbot_segment_guest_room:
       segment_id: '3'
-    furbot_segment_office:
+    input_boolean.furbot_segment_office:
       segment_id: '4'
-    furbot_segment_living_room:
+    input_boolean.furbot_segment_living_room:
       segment_id: '5'
-    furbot_segment_laundry:
+    input_boolean.furbot_segment_laundry:
       segment_id: '6'
-    furbot_segment_kitchen:
+    input_boolean.furbot_segment_kitchen:
       segment_id: '7'
-    furbot_segment_entry:
+    input_boolean.furbot_segment_entry:
       segment_id: '8'
-    furbot_segment_dining_room:
+    input_boolean.furbot_segment_dining_room:
       segment_id: '9'
-    furbot_segment_hallway:
+    input_boolean.furbot_segment_hallway:
       segment_id: '10'
 
 template: