diff --git a/packages/irrigation/main.yaml b/packages/irrigation/main.yaml
index 84d2d21..05837e6 100644
--- a/packages/irrigation/main.yaml
+++ b/packages/irrigation/main.yaml
@@ -2,25 +2,29 @@
 
 # -----------------------------------------------------------------------------
 # Irrigation Sprinkler Groups
-groups:
-  name: Irrigation
-  icon: mdi:water-pump
-  view: true
-  entities:
-    - group.sprinkler_zones_front
-    - group.sprinkler_zones_back
-    - switch.all_areas_schedule
+switch:
+  - platform: group
+    name: Sprinkler Zones Front
+    entities:
+      - switch.front_yard
+      - switch.front_garden
+      - switch.front_street
 
-  name: Sprinkler Zones Front
-  view: false
-  entities:
-    - switch.front_yard
-    - switch.front_garden
-    - switch.front_street
+  - platform: group
+    name: Sprinkler Zones Back
+    entities:
+      - switch.back_yard_grass
+      - switch.back_yard_patio
+      - switch.back_yard_shrubs
 
-  name: Sprinkler Zones Back
-  view: false
-  entities:
-    - switch.back_yard_grass
-    - switch.back_yard_patio
-    - switch.back_yard_shrubs
+  - platform: group
+    name: Irrigation
+    entities:
+      - switch.sprinkler_zones_front
+      - switch.sprinkler_zones_back
+      - switch.all_areas_schedule
+
+homeassistant:
+  customize:
+    switch.irrigation:
+      icon: mdi:water-pump