diff --git a/homeassistant/components/avion/light.py b/homeassistant/components/avion/light.py index ac599f474bc..7df17c2e74e 100644 --- a/homeassistant/components/avion/light.py +++ b/homeassistant/components/avion/light.py @@ -8,8 +8,8 @@ import voluptuous as vol from homeassistant.components.light import ( ATTR_BRIGHTNESS, - COLOR_MODE_BRIGHTNESS, PLATFORM_SCHEMA, + ColorMode, LightEntity, ) from homeassistant.const import ( @@ -73,8 +73,8 @@ def setup_platform( class AvionLight(LightEntity): """Representation of an Avion light.""" - _attr_support_color_mode = COLOR_MODE_BRIGHTNESS - _attr_support_color_modes = {COLOR_MODE_BRIGHTNESS} + _attr_support_color_mode = ColorMode.BRIGHTNESS + _attr_support_color_modes = {ColorMode.BRIGHTNESS} _attr_should_poll = False _attr_assumed_state = True _attr_is_on = True