diff --git a/custom_components/traeger/entity.py b/custom_components/traeger/entity.py
index b36d0bb..958c427 100644
--- a/custom_components/traeger/entity.py
+++ b/custom_components/traeger/entity.py
@@ -61,7 +61,7 @@ def device_info(self):
         }
 
     @property
-    def device_state_attributes(self):
+    def extra_state_attributes(self):
         """Return the state attributes."""
         return {
             "attribution": ATTRIBUTION,