diff --git a/custom_components/varna_public_transport/sensor.py b/custom_components/varna_public_transport/sensor.py
index 72ee31d..10c84dd 100644
--- a/custom_components/varna_public_transport/sensor.py
+++ b/custom_components/varna_public_transport/sensor.py
@@ -106,7 +106,7 @@ def should_poll(self):
         return False
 
     @property
-    def device_state_attributes(self):
+    def extra_state_attributes(self):
         """Return the state attributes."""
         attr = {}
         attr["StopName"] = self._stopname