diff --git a/apps/stages.py b/apps/stages.py
index 83ec406..4374074 100644
--- a/apps/stages.py
+++ b/apps/stages.py
@@ -8,8 +8,9 @@ from pydantic import field_serializer
 
 class EntityState(BaseModel):
     state: bool = True
-    color_temp_kelvin: int
-    brightness: int
+    color_temp_kelvin: int | None = None
+    rgb_color: list[int] | None = None
+    brightness: int | None = None
 
     @field_serializer('state')
     def convert_state(self, val: Any):