diff --git a/motion-sensing-lights.yaml b/motion-sensing-lights.yaml
index 868b7a3..568f3fa 100644
--- a/motion-sensing-lights.yaml
+++ b/motion-sensing-lights.yaml
@@ -162,13 +162,10 @@ action:
             sequence:
               - service: switch.turn_on
                 target: !input entity_target
-  - wait_for_trigger:
-      platform: state
-      entity_id: !input motion_sensors
-      to: "off"
   - if:
       - condition: state
         entity_id: !input motion_sensors
+        match: any
         state: "on"
     then:
       - wait_for_trigger: