diff --git a/homeassistant/components/unifiled/light.py b/homeassistant/components/unifiled/light.py index 2708871bdf2..dd5a5b5290a 100644 --- a/homeassistant/components/unifiled/light.py +++ b/homeassistant/components/unifiled/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 CONF_HOST, CONF_PASSWORD, CONF_PORT, CONF_USERNAME @@ -59,8 +59,8 @@ def setup_platform( class UnifiLedLight(LightEntity): """Representation of an unifiled Light.""" - _attr_color_mode = COLOR_MODE_BRIGHTNESS - _attr_supported_color_modes = {COLOR_MODE_BRIGHTNESS} + _attr_color_mode = ColorMode.BRIGHTNESS + _attr_supported_color_modes = {ColorMode.BRIGHTNESS} def __init__(self, light, api): """Init Unifi LED Light."""