diff --git a/tests/test-ext.yaml b/tests/test-ext.yaml
index 718bb64..80a5c23 100644
--- a/tests/test-ext.yaml
+++ b/tests/test-ext.yaml
@@ -94,7 +94,7 @@ switch:
   - platform: template
     name: AC Display
     lambda: |-
-      if (ac_display_state).state) {
+      if (id(ac_display_state).state) {
         return true;
       } else {
         return false;