diff --git a/query/query_test.go b/query/query_test.go
index 56dc9dfb369ccabc1e34ed6278cabdfbaa98371d..7cf511dcb9405725d0c78833be8cd3fe757f7c4a 100644
--- a/query/query_test.go
+++ b/query/query_test.go
@@ -214,7 +214,7 @@ func TestProcessGraph(t *testing.T) {
 	}
 
 	ch := make(chan error)
-	go ProcessGraph(sg, ch)
+	go ProcessGraph(sg, ch, time.Minute)
 	err = <-ch
 	if err != nil {
 		t.Error(err)
@@ -306,7 +306,7 @@ func TestToJson(t *testing.T) {
 	}
 
 	ch := make(chan error)
-	go ProcessGraph(sg, ch)
+	go ProcessGraph(sg, ch, time.Minute)
 	err = <-ch
 	if err != nil {
 		t.Error(err)
@@ -347,7 +347,7 @@ func TestPreTraverse(t *testing.T) {
 	}
 
 	ch := make(chan error)
-	go ProcessGraph(sg, ch)
+	go ProcessGraph(sg, ch, time.Minute)
 	err = <-ch
 	if err != nil {
 		t.Error(err)