diff --git a/custom_components/luxtronik/model.py b/custom_components/luxtronik/model.py
index 5ed4dc3..d3e0a80 100644
--- a/custom_components/luxtronik/model.py
+++ b/custom_components/luxtronik/model.py
@@ -83,6 +83,7 @@ class LuxtronikEntityDescription(EntityDescription):
     extra_attributes: list[LuxtronikEntityAttributeDescription] = field(
         default_factory=list
     )
+    state_class: str | None = None
 
 
 @dataclass