diff --git a/esphome/airqualitysensor-2.yaml b/esphome/airqualitysensor-2.yaml
index ae6cf48..1edb80c 100644
--- a/esphome/airqualitysensor-2.yaml
+++ b/esphome/airqualitysensor-2.yaml
@@ -99,7 +99,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;
@@ -154,7 +154,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;