diff --git a/include/main.h b/include/main.h
index 3ec571f..5f42a05 100644
--- a/include/main.h
+++ b/include/main.h
@@ -1,11 +1,13 @@
 #ifndef __MAIN_H_
 #define __MAIN_H_
 
+#include "freertos/FreeRTOS.h"
+#include "freertos/queue.h"
 #include "nvs_flash.h"
 
 #define PUMP_COUNT		(4)
 
-static xQueueHandle gpio_evt_queue = NULL;
+static QueueHandle_t gpio_evt_queue = NULL;
 nvs_handle_t config_handle;
 
 extern uint8_t mac[6];