diff --git a/Cargo.toml b/Cargo.toml
index d77831d..8aeed9d 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -17,12 +17,13 @@ paste = "1"
 log = "0.4"
 
 [dev-dependencies]
-datafusion = { version = "47", default-features = false, features = ["nested_expressions"] }
+datafusion = { version = "47", default-features = false , features = ["nested_expressions"]}
 codspeed-criterion-compat = "2.6"
-criterion = "0.5.1"
-clap = "4"
 tokio = { version = "1.43", features = ["full"] }
 
+[patch.crates-io]
+datafusion = { git = "https://github.com/apache/datafusion.git", rev = "5bdaeaf" }
+
 [lints.clippy]
 dbg_macro = "deny"
 print_stdout = "deny"
diff --git a/tests/main.rs b/tests/main.rs
index f591385..250ce90 100644
--- a/tests/main.rs
+++ b/tests/main.rs
@@ -503,8 +503,13 @@ fn test_json_get_utf8() {
     let ColumnarValue::Scalar(sv) = json_get_str
         .invoke_with_args(ScalarFunctionArgs {
             args: args.to_vec(),
+            arg_fields: vec![
+                &Field::new("arg_1", DataType::LargeUtf8, false),
+                &Field::new("arg_2", DataType::LargeUtf8, false),
+                &Field::new("arg_3", DataType::LargeUtf8, false),
+            ],
             number_rows: 1,
-            return_type: &DataType::Utf8,
+            return_field: &Field::new("ret_field", DataType::Utf8, false),
         })
         .unwrap()
     else {
@@ -528,8 +533,13 @@ fn test_json_get_large_utf8() {
     let ColumnarValue::Scalar(sv) = json_get_str
         .invoke_with_args(ScalarFunctionArgs {
             args: args.to_vec(),
+            arg_fields: vec![
+                &Field::new("arg_1", DataType::LargeUtf8, false),
+                &Field::new("arg_2", DataType::LargeUtf8, false),
+                &Field::new("arg_3", DataType::LargeUtf8, false),
+            ],
             number_rows: 1,
-            return_type: &DataType::LargeUtf8,
+            return_field: &Field::new("ret_field", DataType::Utf8, false),
         })
         .unwrap()
     else {