diff --git a/include/sensors/atv.yaml b/include/sensors/atv.yaml
index 875c49d..855b3a5 100644
--- a/include/sensors/atv.yaml
+++ b/include/sensors/atv.yaml
@@ -1,5 +1,5 @@
 - platform: template
   sensors:
     living_room_tv_source:
-      value_template: '{{ states.media_player.stue_2.attributes["source"] }}'
+      value_template: '{{ state_attr('media_player.stue_2','source') | default('unknown') }}'
       friendly_name: 'Living Room TV Source'