diff --git a/src/test/java/com/snowflake/snowpark_test/JavaUDTFSuite.java b/src/test/java/com/snowflake/snowpark_test/JavaUDTFSuite.java
index 3c80daaf..686b4093 100644
--- a/src/test/java/com/snowflake/snowpark_test/JavaUDTFSuite.java
+++ b/src/test/java/com/snowflake/snowpark_test/JavaUDTFSuite.java
@@ -345,6 +345,12 @@ public void partitionBy() {
 
     DataFrame result4 = df.join(tf, map, new Column[] {}, new Column[] {});
     result4.show();
+
+    DataFrame result5 = df.join(tf.call(map), new Column[] {df.col("a")}, new Column[] {df.col("b")});
+    checkAnswer(
+            result5,
+            new Row[] {Row.create("a", null, "{b=2, c=1}"), Row.create("d", null, "{e=1}")},
+            true);
   }
 }