diff --git a/custom_components/integration_blueprint/entity.py b/custom_components/integration_blueprint/entity.py
index 91e50de..0ab5eff 100644
--- a/custom_components/integration_blueprint/entity.py
+++ b/custom_components/integration_blueprint/entity.py
@@ -24,7 +24,7 @@ def device_info(self):
         }
 
     @property
-    def device_state_attributes(self):
+    def extra_state_attributes(self):
         """Return the state attributes."""
         return {
             "attribution": ATTRIBUTION,