diff --git a/homeassistant/components/demo/light.py b/homeassistant/components/demo/light.py index 25a7b46bfb6..21a7dd7440b 100644 --- a/homeassistant/components/demo/light.py +++ b/homeassistant/components/demo/light.py @@ -174,7 +174,7 @@ class DemoLight(LightEntity): return self._brightness @property - def color_mode(self) -> str | None: + def color_mode(self) -> ColorMode | None: """Return the color mode of the light.""" return self._color_mode