diff --git a/hue-dimmer-blueprint.yaml b/hue-dimmer-blueprint.yaml
index 9ee6d27..b35ca59 100644
--- a/hue-dimmer-blueprint.yaml
+++ b/hue-dimmer-blueprint.yaml
@@ -51,24 +51,21 @@ triggers:
       - long_press
       - repeat
   - trigger: state
-    entity_id:
-      - !input button2
+    entity_id: var_button1
     attribute: event_type
     to:
       - short_release
       - long_press
       - repeat
   - trigger: state
-    entity_id:
-      - !input button3
+    entity_id: var_button1
     attribute: event_type
     to:
       - short_release
       - long_press
       - repeat
   - trigger: state
-    entity_id:
-      - !input button4
+    entity_id: var_button1
     attribute: event_type
     to:
       - short_release