diff --git a/src/tree_data_helpers.c b/src/tree_data_helpers.c
index 705b625..168869a 100644
--- a/src/tree_data_helpers.c
+++ b/src/tree_data_helpers.c
@@ -31,6 +31,7 @@
 #include "tree.h"
 #include "tree_data.h"
 #include "tree_data_internal.h"
+#include "tree_edit.h"
 #include "tree_schema.h"
 
 struct lyd_node *