diff --git a/tests/utests/data/test_parser_xml.c b/tests/utests/data/test_parser_xml.c
index e7b16f3..e81a3f0 100644
--- a/tests/utests/data/test_parser_xml.c
+++ b/tests/utests/data/test_parser_xml.c
@@ -24,6 +24,7 @@
 
 #include "../../src/context.h"
 #include "../../src/tree_data_internal.h"
+#include "../../src/tree_schema.h"
 #include "../../src/printer.h"
 #include "../../src/printer_data.h"
 
diff --git a/tests/utests/data/test_printer_xml.c b/tests/utests/data/test_printer_xml.c
index b36d8d5..ad15d16 100644
--- a/tests/utests/data/test_printer_xml.c
+++ b/tests/utests/data/test_printer_xml.c
@@ -18,6 +18,7 @@
 #include <cmocka.h>
 
 #include <stdio.h>
+#include <stdlib.h>
 #include <string.h>
 
 #include "tests/config.h"
@@ -25,6 +26,7 @@
 #include "../../src/context.h"
 #include "../../src/printer.h"
 #include "../../src/printer_data.h"
+#include "../../src/tree_schema.h"
 
 #define BUFSIZE 1024
 char logbuf[BUFSIZE] = {0};
diff --git a/tests/utests/data/test_validation.c b/tests/utests/data/test_validation.c
index 04177a5..836ad25 100644
--- a/tests/utests/data/test_validation.c
+++ b/tests/utests/data/test_validation.c
@@ -23,6 +23,7 @@
 #include <string.h>
 
 #include "../../src/context.h"
+#include "../../src/tree_schema.h"
 #include "../../src/tree_data_internal.h"
 #include "../../src/printer.h"
 #include "../../src/printer_data.h"