diff --git a/tests/src/test_context.c b/tests/src/test_context.c
index cc5adfa..d7c75fb 100644
--- a/tests/src/test_context.c
+++ b/tests/src/test_context.c
@@ -21,6 +21,8 @@
 #include "../../src/parser_yang.c"
 #include "../../src/context.c"
 #include "../../src/tree_schema_helpers.c"
+#include "../../src/tree_schema_free.c"
+#include "../../src/tree_schema_compile.c"
 #include "../../src/tree_schema.c"
 
 #include <stdarg.h>
diff --git a/tests/src/test_hash_table.c b/tests/src/test_hash_table.c
index 7411206..2f10f0a 100644
--- a/tests/src/test_hash_table.c
+++ b/tests/src/test_hash_table.c
@@ -23,6 +23,8 @@
 #include "../../src/parser_yang.c"
 #include "../../src/context.c"
 #include "../../src/tree_schema_helpers.c"
+#include "../../src/tree_schema_free.c"
+#include "../../src/tree_schema_compile.c"
 #include "../../src/tree_schema.c"
 
 #include <stdarg.h>
diff --git a/tests/src/test_parser_yang.c b/tests/src/test_parser_yang.c
index 23144dc..9ebfbec 100644
--- a/tests/src/test_parser_yang.c
+++ b/tests/src/test_parser_yang.c
@@ -20,6 +20,8 @@
 #include "../../src/parser_yang.c"
 #include "../../src/context.c"
 #include "../../src/tree_schema_helpers.c"
+#include "../../src/tree_schema_free.c"
+#include "../../src/tree_schema_compile.c"
 #include "../../src/tree_schema.c"
 
 #include <stdarg.h>
diff --git a/tests/src/test_tree_schema_compile.c b/tests/src/test_tree_schema_compile.c
index a0d5cd1..bc4169c 100644
--- a/tests/src/test_tree_schema_compile.c
+++ b/tests/src/test_tree_schema_compile.c
@@ -18,6 +18,8 @@
 #include "../../src/xpath.c"
 #include "../../src/parser_yang.c"
 #include "../../src/tree_schema_helpers.c"
+#include "../../src/tree_schema_free.c"
+#include "../../src/tree_schema_compile.c"
 #include "../../src/tree_schema.c"
 #include "../../src/context.c"
 #include "../../src/hash_table.c"
diff --git a/tests/src/test_tree_schema_helpers.c b/tests/src/test_tree_schema_helpers.c
index 580d488..29b7e8c 100644
--- a/tests/src/test_tree_schema_helpers.c
+++ b/tests/src/test_tree_schema_helpers.c
@@ -17,6 +17,7 @@
 #include "../../src/set.c"
 #include "../../src/parser_yang.c"
 #include "../../src/tree_schema.c"
+#include "../../src/tree_schema_free.c"
 #include "../../src/tree_schema_helpers.c"
 #include "../../src/hash_table.c"
 #include "../../src/xpath.c"