diff --git a/src/OTASettingsService.cpp b/src/OTASettingsService.cpp
index a74d34f..7de85b4 100644
--- a/src/OTASettingsService.cpp
+++ b/src/OTASettingsService.cpp
@@ -39,7 +39,9 @@ void OTASettingsService::writeToJsonObject(JsonObject& root) {
 
 void OTASettingsService::configureArduinoOTA() {
   if (_arduinoOTA){
+#if defined(ESP_PLATFORM)
     _arduinoOTA->end();
+#endif
     delete _arduinoOTA;
     _arduinoOTA = nullptr;
   }
diff --git a/src/main.cpp b/src/main.cpp
index 1aa3266..bdae650 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -10,7 +10,6 @@
 #endif
 
 #include <FS.h>
-#include <esp_wifi.h>
 #include <SecuritySettingsService.h>
 #include <WiFiSettingsService.h>
 #include <APSettingsService.h>