diff --git a/publisher.py b/publisher.py
index 3347909..3f23f6c 100644
--- a/publisher.py
+++ b/publisher.py
@@ -338,11 +338,11 @@ class Publisher:
         keys = []
         if LightSensor.ONOFF in version:
             keys.extend(["on", "reachable"])
-        if LightSensor.BRIGHTNESS_LIGHT in version:
+        if LightSensor.BRIGHTNESS in version:
             keys.extend(["brightness"])
-        if LightSensor.TEMPERATURE_LIGHT in version:
+        if LightSensor.TEMPERATURE in version:
             keys.extend(["ct"])
-        if LightSensor.COLOUR_LIGHT in version:
+        if LightSensor.COLOUR in version:
             keys.extend(["hue", "saturation", "xy"])
 
         return Payload(self, "light", keys=keys,