diff --git a/aql/convertQuery.go b/aql/convertQuery.go
index ba5999dd1aa658684b26498464a8beaad51a7a82..b85757c8018f9a9f2ac7ef3d6cf24fc6c85f563f 100644
--- a/aql/convertQuery.go
+++ b/aql/convertQuery.go
@@ -3,7 +3,9 @@
 // © Copyright Utrecht University (Department of Information and Computing Sciences)
 // */
 
-// package aql
+package aql
+
+import "git.science.uu.nl/graphpolaris/query-conversion/entity"
 
 // import (
 // 	"errors"
diff --git a/cypher/convertQuery.go b/cypher/convertQuery.go
index 397966b5ccedf1fe4b2acf48f197321f70bc2307..310639efa6c57f2ff72dfd29c6c886391fa459d7 100644
--- a/cypher/convertQuery.go
+++ b/cypher/convertQuery.go
@@ -120,7 +120,7 @@ func createReturnStatement(JSONQuery *entity.IncomingQueryJSON, parts entity.Que
 		retType = ";nodelink"
 	}
 
-	retStatement = retStatement + "\n" + fmt.Sprintf("LIMIT %v", JSONQuery.Limit)
+	retStatement = retStatement + "\n" + fmt.Sprintf("LIMIT %v", JSONQuery.Limit) + retType
 
 	return &retStatement, nil
 }