diff --git a/occupancy_controlled_lights.yaml b/occupancy_controlled_lights.yaml
index 15c4d67..a224060 100644
--- a/occupancy_controlled_lights.yaml
+++ b/occupancy_controlled_lights.yaml
@@ -81,9 +81,12 @@ variables:
 
 trigger:
   - trigger: state
-    entity_id: "{{ sensors }}"
+    domain: binary_sensor
+    device_class: occupancy
 
-condition: []
+condition:
+  - condition: template
+    value_template: "{{ trigger.entity_id in sensors }}"
 
 action:
   - choose: