diff --git a/nspanel_esphome.yaml b/nspanel_esphome.yaml
index c394a64..27ece33 100644
--- a/nspanel_esphome.yaml
+++ b/nspanel_esphome.yaml
@@ -1171,14 +1171,20 @@ text_sensor:
             if (page=="boot")
               {
                 // Detect display mode - Need review
-                std::string display_mode_str = doc["display_mode"];
-                ESP_LOGV("text_sensor.nspanelevent", "display_mode: %s", display_mode_str.c_str());
-                float display_mode_float = stof(display_mode_str);
-                if (display_mode_float > 0) id(display_mode) = int(display_mode_float);
+                if (doc.containsKey("display_mode"))
+                  {
+                    std::string display_mode_str = doc["display_mode"];
+                    ESP_LOGV("text_sensor.nspanelevent", "display_mode: %s", display_mode_str.c_str());
+                    float display_mode_float = stof(display_mode_str);
+                    if (display_mode_float > 0) id(display_mode) = int(display_mode_float);
+                  }
 
                 // Detect TFT version
-                std::string version_tmp = doc["version"];
-                id(version_tft) = version_tmp;
+                if (doc.containsKey("version"))
+                  {
+                    std::string version_tmp = doc["version"];
+                    id(version_tft) = version_tmp;
+                  }
                 id(check_versions).execute();
 
                 // Detect timeout