diff --git a/src/test/java/example/FindAnyOnePathTest.java b/src/test/java/example/FindAnyOnePathTest.java
index 84dffba..4ad3878 100644
--- a/src/test/java/example/FindAnyOnePathTest.java
+++ b/src/test/java/example/FindAnyOnePathTest.java
@@ -59,7 +59,7 @@ void findAnyOnePath() {
             // language=cypher
             var records = session.run("match (start:Source)" +
                     "match (end: Sink) " +
-                    "call bytecodedl.findOnePath(start, end, 10, 'insn') yield path return path").list();
+                    "call bytecodedl.findOnePath(start, end, 10) yield path return path").list();
 
             for(Record record : records){
                 //System.out.println(record.get("path"));
@@ -80,7 +80,7 @@ void biFindOnePath() {
             // language=cypher
             var records = session.run("match (start:Source)" +
                     "match (end: Sink1) " +
-                    "call bytecodedl.biFindOnePath(start, end, 10, 'insn') yield path return path").list();
+                    "call bytecodedl.biFindOnePath(start, end, 10) yield path return path").list();
 
             for(Record record : records){
                 //System.out.println(record.get("path"));