diff --git a/esphome/airqualitysensor-1.yaml b/esphome/airqualitysensor-1.yaml
index 578a329..17b60a9 100644
--- a/esphome/airqualitysensor-1.yaml
+++ b/esphome/airqualitysensor-1.yaml
@@ -93,7 +93,7 @@ script:
 
           std::string quality = id(iaq_reading).state;
           float lux = id(ambient_light).state;
-          std::string mode = id(led_mode).state;
+          std::string mode = id(led_mode).current_option();
 
           float brightness = 0.15;
           float r = 0.0, g = 0.0, b = 0.0;
@@ -148,7 +148,7 @@ script:
           }
           
           std::string quality = id(iaq_reading).state;
-          std::string mode = id(led_mode).state;
+          std::string mode = id(led_mode).current_option();
           float lux = id(ambient_light).state;
           
           float r = 0.0, g = 0.0, b = 0.0;