diff --git a/src/edge.c b/src/edge.c
index 19f5645..dedd6a0 100644
--- a/src/edge.c
+++ b/src/edge.c
@@ -22,6 +22,7 @@
 #include "r3.h"
 #include "r3_str.h"
 #include "str_array.h"
+#include "zmalloc.h"
 
 edge * r3_edge_create(char * pattern, int pattern_len, node * child) {
     edge * e = (edge*) zmalloc( sizeof(edge) );
diff --git a/tests/bench_str.csv b/tests/bench_str.csv
index b0e5c9f..603fd34 100644
--- a/tests/bench_str.csv
+++ b/tests/bench_str.csv
@@ -438,3 +438,5 @@
 1400607695,13583203.56
 1400607698,13630627.45
 1400607700,13972490.11
+1400655532,6648299.49
+1400655542,6685928.45
diff --git a/tests/check_tree.c b/tests/check_tree.c
index 6565f3b..718d711 100644
--- a/tests/check_tree.c
+++ b/tests/check_tree.c
@@ -5,6 +5,7 @@
 #include "r3.h"
 #include "r3_str.h"
 #include "str_array.h"
+#include "zmalloc.h"
 #include "bench.h"