diff --git a/custom_components/lightwave2/light.py b/custom_components/lightwave2/light.py
index b856860..9ce62d5 100644
--- a/custom_components/lightwave2/light.py
+++ b/custom_components/lightwave2/light.py
@@ -5,7 +5,7 @@
 except ImportError:
     from homeassistant.components.light import Light as LightEntity
 from homeassistant.components.light import (
-    ATTR_BRIGHTNESS, SUPPORT_BRIGHTNESS)
+    ATTR_BRIGHTNESS, COLOR_MODE_BRIGHTNESS)
 from homeassistant.core import callback
 from .const import DOMAIN
 
@@ -79,9 +79,14 @@ def async_update_callback(self, **kwargs):
         self.async_schedule_update_ha_state(True)
 
     @property
-    def supported_features(self):
+    def supported_color_modes(self):
         """Flag supported features."""
-        return SUPPORT_BRIGHTNESS
+        return {COLOR_MODE_BRIGHTNESS}
+
+    @property
+    def color_mode(self):
+        """Flag supported features."""
+        return COLOR_MODE_BRIGHTNESS
 
     @property
     def should_poll(self):
diff --git a/custom_components/lightwave2/manifest.json b/custom_components/lightwave2/manifest.json
index 309315d..7f9ceae 100644
--- a/custom_components/lightwave2/manifest.json
+++ b/custom_components/lightwave2/manifest.json
@@ -1,7 +1,7 @@
 {
   "domain": "lightwave2",
   "name": "Lightwave v2 (Smart Series) custom component",
-  "version": "3.4.0",
+  "version": "3.4.1",
   "documentation": "https://github.com/bigbadblunt/homeassistant-lightwave2",
   "dependencies": [],
   "codeowners": ["@bigbadblunt"],