tree schema MAINTENANCE split source files to keep smaller files to work with
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>