diff --git a/src/tree_schema_free.c b/src/tree_schema_free.c
index cea7784..09ce99d 100644
--- a/src/tree_schema_free.c
+++ b/src/tree_schema_free.c
@@ -22,6 +22,7 @@
 #include "tree.h"
 #include "tree_data.h"
 #include "tree_data_internal.h"
+#include "tree_edit.h"
 #include "tree_schema.h"
 #include "tree_schema_internal.h"
 #include "xml.h"