diff --git a/examples/GatewayESP8266/GatewayESP8266.ino b/examples/GatewayESP8266/GatewayESP8266.ino
index f4f6fd8..4e7f38d 100644
--- a/examples/GatewayESP8266/GatewayESP8266.ino
+++ b/examples/GatewayESP8266/GatewayESP8266.ino
@@ -126,11 +126,11 @@
 #define MY_DEFAULT_TX_LED_PIN  16  // the PCB, on board LED
 
 #if defined(MY_USE_UDP)
-  #include <WiFiUDP.h>
-#else
-  #include <ESP8266WiFi.h>
+  #include <WiFiUdp.h>
 #endif
 
+#include <ESP8266WiFi.h>
+
 #include <MySensors.h>
 
 void setup() {