diff --git a/hue-dimmer-blueprint.yaml b/hue-dimmer-blueprint.yaml
index 14ddd0e..6c33e30 100644
--- a/hue-dimmer-blueprint.yaml
+++ b/hue-dimmer-blueprint.yaml
@@ -3,11 +3,12 @@ blueprint:
   description: Turn a light on based on detected motion
   domain: automation
   input:
-    button1:
-    button2:
-    button3:
-    button4:
+    switch_device:
+        selector:
+            device:
     button1_short_press:
+        selector:
+            action:
 triggers:
   - trigger: state
     entity_id: