diff --git a/custom_components/youtube/sensor.py b/custom_components/youtube/sensor.py
index 1685959..917e126 100644
--- a/custom_components/youtube/sensor.py
+++ b/custom_components/youtube/sensor.py
@@ -118,7 +118,7 @@ def icon(self):
         return ICON
 
     @property
-    def device_state_attributes(self):
+    def extra_state_attributes(self):
         """Attributes."""
         return {'url': self.url,
                 'published': self.published,