diff --git a/tools/test_apps/protocols/mdns/main/mdns_test.c b/tools/test_apps/protocols/mdns/main/mdns_test.c
index b47e67cc1..d82229f1d 100644
--- a/tools/test_apps/protocols/mdns/main/mdns_test.c
+++ b/tools/test_apps/protocols/mdns/main/mdns_test.c
@@ -4,6 +4,8 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 #include <string.h>
+#include "freertos/FreeRTOS.h"
+#include "freertos/task.h"
 #include "mdns.h"
 #include "esp_log.h"
 #include "esp_netif.h"