diff --git a/HyperionRGB/WrapperWebconfig.h b/HyperionRGB/WrapperWebconfig.h
index 0241e47..59bdcb0 100644
--- a/HyperionRGB/WrapperWebconfig.h
+++ b/HyperionRGB/WrapperWebconfig.h
@@ -2,7 +2,11 @@
 #define WrapperWebconfig_h
 #include "BaseHeader.h"
 
-#include <ESP8266WebServer.h>
+#if defined(ESP8266)
+  #include <ESP8266WebServer.h>
+#elif defined(ESP32)
+  #include <ESP32WebServer.h>
+#endif
 #include <LinkedList.h>
 
 class SelectEntryBase {
diff --git a/HyperionRGB/WrapperWiFi.h b/HyperionRGB/WrapperWiFi.h
index d20e07f..5bb3072 100644
--- a/HyperionRGB/WrapperWiFi.h
+++ b/HyperionRGB/WrapperWiFi.h
@@ -2,8 +2,14 @@
 #define WrapperWiFi_h
 
 #include "BaseHeader.h"
-#include <ESP8266WiFi.h>
-#include <ESP8266mDNS.h>
+
+#if defined(ESP8266)
+  #include <ESP8266WiFi.h>
+  #include <ESP8266mDNS.h>
+#elif defined(ESP32)
+  #include <WiFi.h>
+  #include <ESPmDNS.h>
+#endif
 
 class WrapperWiFi {
   public: