From 9c4244391045cd1d88bb4bbbb7b9837c09e7e857 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Sat, 23 Apr 2022 21:43:08 +0200
Subject: [PATCH] Use ColorMode enum in smarttub (#70536)

---
 homeassistant/components/smarttub/light.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

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."""