diff --git a/main/wifi.c b/main/wifi.c
--- a/main/wifi.c
+++ b/main/wifi.c
@@ -9,7 +9,9 @@
 #include "lwip/err.h"
 #include "lwip/sys.h"
 #include "display.h"
-
+#include "freertos/FreeRTOS.h"
+#include "freertos/task.h"
+#include "freertos/event_groups.h"
 
 // Private variables