diff --git a/main/main.c b/main/main.c
index eac3bba..e11b426 100644
--- a/main/main.c
+++ b/main/main.c
@@ -12,6 +12,10 @@
 #include "freertos/task.h"
 #include "esp_system.h"
 #include "esp_spi_flash.h"
+#include "esp_idf_version.h"
+#if ESP_IDF_VERSION >= ESP_IDF_VERSION_VAL(4, 4, 0)
+#include "esp_chip_info.h"
+#endif
 
 #include "RustApi.h"