diff --git a/homeassistant/components/group/light.py b/homeassistant/components/group/light.py index c940a21e98a..b9741085c2d 100644 --- a/homeassistant/components/group/light.py +++ b/homeassistant/components/group/light.py @@ -27,10 +27,9 @@ from homeassistant.components.light import ( ATTR_WHITE, ATTR_WHITE_VALUE, ATTR_XY_COLOR, - COLOR_MODE_BRIGHTNESS, - COLOR_MODE_ONOFF, PLATFORM_SCHEMA, SUPPORT_WHITE_VALUE, + ColorMode, LightEntity, LightEntityFeature, ) @@ -286,10 +285,10 @@ class LightGroup(GroupEntity, LightEntity): if all_color_modes: # Report the most common color mode, select brightness and onoff last color_mode_count = Counter(itertools.chain(all_color_modes)) - if COLOR_MODE_ONOFF in color_mode_count: - color_mode_count[COLOR_MODE_ONOFF] = -1 - if COLOR_MODE_BRIGHTNESS in color_mode_count: - color_mode_count[COLOR_MODE_BRIGHTNESS] = 0 + if ColorMode.ONOFF in color_mode_count: + color_mode_count[ColorMode.ONOFF] = -1 + if ColorMode.BRIGHTNESS in color_mode_count: + color_mode_count[ColorMode.BRIGHTNESS] = 0 self._attr_color_mode = color_mode_count.most_common(1)[0][0] self._attr_supported_color_modes = None