diff --git a/sql/core/src/test/scala/org/apache/spark/sql/execution/benchmark/TPCDSQueryBenchmark.scala b/sql/core/src/test/scala/org/apache/spark/sql/execution/benchmark/TPCDSQueryBenchmark.scala
index 721997d84e1a..1ff6122906d1 100644
--- a/sql/core/src/test/scala/org/apache/spark/sql/execution/benchmark/TPCDSQueryBenchmark.scala
+++ b/sql/core/src/test/scala/org/apache/spark/sql/execution/benchmark/TPCDSQueryBenchmark.scala
@@ -74,7 +74,8 @@ object TPCDSQueryBenchmark extends SqlBasedBenchmark with Logging {
     tables.map { tableName =>
       spark.sql(s"DROP TABLE IF EXISTS $tableName")
       val options = Map("path" -> s"$dataLocation/$tableName")
-      spark.catalog.createTable(tableName, "parquet", tableColumns(tableName), options)
+      val format = spark.conf.get("spark.sql.sources.default")
+      spark.catalog.createTable(tableName, format, tableColumns(tableName), options)
       // Recover partitions but don't fail if a table is not partitioned.
       Try {
         spark.sql(s"ALTER TABLE $tableName RECOVER PARTITIONS")