diff --git a/src/main.cpp b/src/main.cpp
index 8194f8a..e35fcb6 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -90,7 +90,7 @@ static void ledTask( void * parameter )
 
     uint32_t start_time = micros();
 
-    if(WiFi.status() == WL_CONNECTED) {
+    if((WiFi.status() == WL_CONNECTED) || (WiFi.getMode() == WIFI_MODE_AP)) {
       udpProto.loop();
     }