diff --git a/src/test/scala/com/snowflake/snowpark_test/UDTFSuite.scala b/src/test/scala/com/snowflake/snowpark_test/UDTFSuite.scala
index de6464be..9c363d86 100644
--- a/src/test/scala/com/snowflake/snowpark_test/UDTFSuite.scala
+++ b/src/test/scala/com/snowflake/snowpark_test/UDTFSuite.scala
@@ -100,8 +100,7 @@ class UDTFSuite extends TestData {
         Seq(
           Row("w1 w2", "g1", "w2", 1),
           Row("w1 w2", "g1", "w1", 1),
-          Row("w1 w1 w1", "g2", "w1", 3)),
-        false)
+          Row("w1 w1 w1", "g2", "w1", 3)))
     } finally {
       runQuery(s"drop function if exists $funcName(STRING)", session)
     }
@@ -146,8 +145,7 @@ class UDTFSuite extends TestData {
             |""".stripMargin)
       checkAnswer(
         df1,
-        Seq(Row("w3", 6), Row("w2", 4), Row("w1", 2), Row("w3", 6), Row("w2", 4), Row("w1", 2)),
-        false)
+        Seq(Row("w3", 6), Row("w2", 4), Row("w1", 2), Row("w3", 6), Row("w2", 4), Row("w1", 2)))
 
       // Call the UDTF with funcName and named parameters, result should be the same
       val df2 = session
@@ -163,8 +161,7 @@ class UDTFSuite extends TestData {
             |""".stripMargin)
       checkAnswer(
         df2,
-        Seq(Row(6, "w3"), Row(4, "w2"), Row(2, "w1"), Row(6, "w3"), Row(4, "w2"), Row(2, "w1")),
-        false)
+        Seq(Row(6, "w3"), Row(4, "w2"), Row(2, "w1"), Row(6, "w3"), Row(4, "w2"), Row(2, "w1")))
 
       // scalastyle:off
       // Use UDTF with table join
@@ -200,8 +197,7 @@ class UDTFSuite extends TestData {
           Row(null, null, "g2", 1),
           Row(null, null, "w2", 1),
           Row(null, null, "g1", 1),
-          Row(null, null, "w1", 4)),
-        false)
+          Row(null, null, "w1", 4)))
 
       // Use UDTF with table function + over partition
       val df4 = session.sql(
@@ -218,8 +214,7 @@ class UDTFSuite extends TestData {
           Row("w1 w1 w1", "g2", "g2", 1),
           Row("w1 w1 w1", "g2", "w1", 3),
           Row(null, "g2", "g2", 1),
-          Row(null, "g2", "w1", 3)),
-        false)
+          Row(null, "g2", "w1", 3)))
     } finally {
       runQuery(s"drop function if exists $funcName(VARCHAR,VARCHAR)", session)
     }
@@ -246,7 +241,7 @@ class UDTFSuite extends TestData {
           """root
             | |--C1: Long (nullable = true)
             |""".stripMargin)
-      checkAnswer(df1, Seq(Row(10), Row(11), Row(12), Row(13), Row(14)), false)
+      checkAnswer(df1, Seq(Row(10), Row(11), Row(12), Row(13), Row(14)))
 
       val df2 = session.tableFunction(TableFunction(funcName), lit(20), lit(5))
       assert(
@@ -254,7 +249,7 @@ class UDTFSuite extends TestData {
           """root
             | |--C1: Long (nullable = true)
             |""".stripMargin)
-      checkAnswer(df2, Seq(Row(20), Row(21), Row(22), Row(23), Row(24)), false)
+      checkAnswer(df2, Seq(Row(20), Row(21), Row(22), Row(23), Row(24)))
 
       val df3 = session
         .tableFunction(tableFunction, Map("arg1" -> lit(30), "arg2" -> lit(5)))
@@ -263,7 +258,7 @@ class UDTFSuite extends TestData {
           """root
             | |--C1: Long (nullable = true)
             |""".stripMargin)
-      checkAnswer(df3, Seq(Row(30), Row(31), Row(32), Row(33), Row(34)), false)
+      checkAnswer(df3, Seq(Row(30), Row(31), Row(32), Row(33), Row(34)))
     } finally {
       runQuery(s"drop function if exists $funcName(NUMBER, NUMBER)", session)
     }
@@ -378,7 +373,7 @@ class UDTFSuite extends TestData {
           """root
             | |--C1: Long (nullable = true)
             |""".stripMargin)
-      checkAnswer(df1, Seq(Row(10), Row(11), Row(20), Row(21), Row(22), Row(23)), false)
+      checkAnswer(df1, Seq(Row(10), Row(11), Row(20), Row(21), Row(22), Row(23)))
 
       val df2 = session.tableFunction(TableFunction(funcName), sourceDF("b"), sourceDF("c"))
       assert(
@@ -386,7 +381,7 @@ class UDTFSuite extends TestData {
           """root
             | |--C1: Long (nullable = true)
             |""".stripMargin)
-      checkAnswer(df2, Seq(Row(100), Row(101), Row(200), Row(201), Row(202), Row(203)), false)
+      checkAnswer(df2, Seq(Row(100), Row(101), Row(200), Row(201), Row(202), Row(203)))
 
       // Check table function with df column arguments as Map
       val sourceDF2 = Seq(30).toDF("a")
@@ -397,7 +392,7 @@ class UDTFSuite extends TestData {
           """root
             | |--C1: Long (nullable = true)
             |""".stripMargin)
-      checkAnswer(df3, Seq(Row(30), Row(31), Row(32), Row(33), Row(34)), false)
+      checkAnswer(df3, Seq(Row(30), Row(31), Row(32), Row(33), Row(34)))
 
       // Check table function with nested functions on df column
       val df4 = session.tableFunction(tableFunction, abs(ceil(sourceDF("a"))), lit(2))
@@ -406,7 +401,7 @@ class UDTFSuite extends TestData {
           """root
             | |--C1: Long (nullable = true)
             |""".stripMargin)
-      checkAnswer(df4, Seq(Row(10), Row(11), Row(20), Row(21)), false)
+      checkAnswer(df4, Seq(Row(10), Row(11), Row(20), Row(21)))
 
       // Check result df column filtering with duplicate column names
       val sourceDF3 = Seq(30).toDF("C1")
@@ -420,7 +415,7 @@ class UDTFSuite extends TestData {
             """root
               | |--C1: Long (nullable = true)
               |""".stripMargin)
-        checkAnswer(df, Seq(Row(30), Row(31), Row(32), Row(33), Row(34)), false)
+        checkAnswer(df, Seq(Row(30), Row(31), Row(32), Row(33), Row(34)))
       }
     } finally {
       runQuery(s"drop function if exists $funcName(NUMBER, NUMBER)", session)
@@ -1867,16 +1862,13 @@ class UDTFSuite extends TestData {
     val tableFunction3 = session.udtf.registerTemporary(new ReturnManyColumns(3))
     val df3 = session.tableFunction(tableFunction3, lit(10))
     assert(df3.schema.length == 3)
-    checkAnswer(df3, Seq(Row(11, 12, 13), Row(1, 2, 3)), false)
+    checkAnswer(df3, Seq(Row(11, 12, 13), Row(1, 2, 3)))
 
     // Test UDTF return 100 columns
     val tableFunction100 = session.udtf.registerTemporary(new ReturnManyColumns(100))
     val df100 = session.tableFunction(tableFunction100, lit(20))
     assert(df100.schema.length == 100)
-    checkAnswer(
-      df100,
-      Seq(Row.fromArray((21 to 120).toArray), Row.fromArray((1 to 100).toArray)),
-      false)
+    checkAnswer(df100, Seq(Row.fromArray((21 to 120).toArray), Row.fromArray((1 to 100).toArray)))
 
     // Test UDTF return 200 columns
     val tableFunction200 = session.udtf.registerTemporary(new ReturnManyColumns(200))
@@ -1884,8 +1876,7 @@ class UDTFSuite extends TestData {
     assert(df200.schema.length == 200)
     checkAnswer(
       df200,
-      Seq(Row.fromArray((101 to 300).toArray), Row.fromArray((1 to 200).toArray)),
-      false)
+      Seq(Row.fromArray((101 to 300).toArray), Row.fromArray((1 to 200).toArray)))
   }
 
   test("test output type: basic types") {