diff --git a/src/OtaHandler.cpp b/src/OtaHandler.cpp
index b2179db..84c2080 100644
--- a/src/OtaHandler.cpp
+++ b/src/OtaHandler.cpp
@@ -2,6 +2,8 @@
 
 #ifndef ESP8266 // Only for esp32 based mcu
 
+#include "AirGradient.h"
+
 void OtaHandler::setHandlerCallback(OtaHandlerCallback_t callback) { _callback = callback; }
 
 void OtaHandler::updateFirmwareIfOutdated(String deviceId) {