diff --git a/src/tree_schema_helpers.c b/src/tree_schema_helpers.c
index bb6c89d..2ed6344 100644
--- a/src/tree_schema_helpers.c
+++ b/src/tree_schema_helpers.c
@@ -35,6 +35,7 @@
 #include "set.h"
 #include "tree.h"
 #include "tree_data.h"
+#include "tree_edit.h"
 #include "tree_schema.h"
 #include "tree_schema_internal.h"