diff --git a/hue-dimmer-switch.yaml b/hue-dimmer-switch.yaml
index a54a36a..c54d90f 100644
--- a/hue-dimmer-switch.yaml
+++ b/hue-dimmer-switch.yaml
@@ -74,7 +74,7 @@ actions:
             metadata: {}
             data: {}
             target:
-              entity_id: !input remote_light
+              entity_id: !input light_target
       - conditions:
           - condition: trigger
             id:
@@ -88,7 +88,7 @@ actions:
                     brightness_step_pct: 3
                     transition: 1
                   target:
-                    entity_id: !input remote_light
+                    entity_id: !input light_target
               until:
                 - condition: trigger
                   id:
@@ -106,7 +106,7 @@ actions:
                     transition: 1
                     brightness_step_pct: -3
                   target:
-                    entity_id: !input remote_light
+                    entity_id: !input light_target
               until:
                 - condition: trigger
                   id:
@@ -119,6 +119,6 @@ actions:
           - action: script.cycle_between_light_profiles
             metadata: {}
             data:
-              light: !input remote_light
+              light: !input light_target
               helper: !input number_helper
 mode: restart