diff --git a/src/tree_data_free.c b/src/tree_data_free.c
index 53d80ca..192f63e 100644
--- a/src/tree_data_free.c
+++ b/src/tree_data_free.c
@@ -17,8 +17,10 @@
 
 #include "common.h"
 #include "hash_table.h"
+#include "log.h"
 #include "plugins_types.h"
 #include "tree.h"
+#include "tree_data.h"
 #include "tree_data_internal.h"
 #include "tree_schema.h"