diff --git a/homeassistant/components/smarttub/light.py b/homeassistant/components/smarttub/light.py index 2cdcefb372b..accb105e88e 100644 --- a/homeassistant/components/smarttub/light.py +++ b/homeassistant/components/smarttub/light.py @@ -4,9 +4,9 @@ from smarttub import SpaLight from homeassistant.components.light import ( ATTR_BRIGHTNESS, ATTR_EFFECT, - COLOR_MODE_BRIGHTNESS, EFFECT_COLORLOOP, SUPPORT_EFFECT, + ColorMode, LightEntity, ) from homeassistant.config_entries import ConfigEntry @@ -43,8 +43,8 @@ async def async_setup_entry( class SmartTubLight(SmartTubEntity, LightEntity): """A light on a spa.""" - _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, coordinator, light): """Initialize the entity."""