diff --git a/custom_components/browser_mod/binary_sensor.py b/custom_components/browser_mod/binary_sensor.py
index 1633380..543fcf2 100644
--- a/custom_components/browser_mod/binary_sensor.py
+++ b/custom_components/browser_mod/binary_sensor.py
@@ -50,7 +50,7 @@ class BrowserModSensor(BrowserModEntity):
         return DEVICE_CLASS_MOTION
 
     @property
-    def device_state_attributes(self):
+    def extra_state_attributes(self):
         return {
             "type": "browser_mod",
             "last_seen": self.last_seen,
diff --git a/custom_components/browser_mod/camera.py b/custom_components/browser_mod/camera.py
index b45cc4b..84aa2d4 100644
--- a/custom_components/browser_mod/camera.py
+++ b/custom_components/browser_mod/camera.py
@@ -37,7 +37,7 @@ class BrowserModCamera(Camera, BrowserModEntity):
         return base64.b64decode(self.data.split(",")[-1])
 
     @property
-    def device_state_attributes(self):
+    def extra_state_attributes(self):
         return {
             "type": "browser_mod",
             "deviceID": self.deviceID,
diff --git a/custom_components/browser_mod/light.py b/custom_components/browser_mod/light.py
index e0b5bde..d1305e3 100644
--- a/custom_components/browser_mod/light.py
+++ b/custom_components/browser_mod/light.py
@@ -40,7 +40,7 @@ class BrowserModLight(LightEntity, BrowserModEntity):
         return not self.data.get("blackout", False)
 
     @property
-    def device_state_attributes(self):
+    def extra_state_attributes(self):
         return {
             "type": "browser_mod",
             "deviceID": self.deviceID,
diff --git a/custom_components/browser_mod/media_player.py b/custom_components/browser_mod/media_player.py
index 6db4c21..dbf7f4e 100644
--- a/custom_components/browser_mod/media_player.py
+++ b/custom_components/browser_mod/media_player.py
@@ -42,7 +42,7 @@ class BrowserModPlayer(MediaPlayerEntity, BrowserModEntity):
         self.schedule_update_ha_state()
 
     @property
-    def device_state_attributes(self):
+    def extra_state_attributes(self):
         return {
             "type": "browser_mod",
             "deviceID": self.deviceID,
diff --git a/custom_components/browser_mod/sensor.py b/custom_components/browser_mod/sensor.py
index 596b525..5e7630a 100644
--- a/custom_components/browser_mod/sensor.py
+++ b/custom_components/browser_mod/sensor.py
@@ -33,7 +33,7 @@ class BrowserModSensor(BrowserModEntity):
         return len(self.connection.connection)
 
     @property
-    def device_state_attributes(self):
+    def extra_state_attributes(self):
         return {
             "type": "browser_mod",
             "last_seen": self.last_seen,