diff --git a/custom_components/mpk_lodz/sensor.py b/custom_components/mpk_lodz/sensor.py
index 56bd918..71da761 100644
--- a/custom_components/mpk_lodz/sensor.py
+++ b/custom_components/mpk_lodz/sensor.py
@@ -78,7 +78,7 @@ def unit_of_measurement(self):
         return None
 
     @property
-    def device_state_attributes(self):
+    def extra_state_attributes(self):
         attr = dict()
         attr['stop_name'] = self._real_stop_name
         if self._departures is not None: