diff --git a/configuration.yaml b/configuration.yaml
index 6d22e81..30fdf67 100644
--- a/configuration.yaml
+++ b/configuration.yaml
@@ -121,10 +121,10 @@ zeroconf:
 # discovery:
 
 # Setup Includes for UI
-automation: !include include/automations.yaml
-scene: !include include/scenes.yaml
-script: !include include/scripts.yaml
-zone: !include include/zones.yaml
+automation ui: !include include/automations.yaml
+scene ui: !include include/scenes.yaml
+script ui: !include include/scripts.yaml
+zone ui: !include include/zones.yaml
 
 # Setup InfluxDB Logging
 influxdb: