diff --git a/src/main/scala/com/databricks/spark/sql/perf/tpcds/TPCDS.scala b/src/main/scala/com/databricks/spark/sql/perf/tpcds/TPCDS.scala
index 85ab4f8..64c6a6a 100644
--- a/src/main/scala/com/databricks/spark/sql/perf/tpcds/TPCDS.scala
+++ b/src/main/scala/com/databricks/spark/sql/perf/tpcds/TPCDS.scala
@@ -19,19 +19,23 @@ package com.databricks.spark.sql.perf.tpcds
 import scala.collection.mutable
 
 import com.databricks.spark.sql.perf._
+import org.apache.spark.SparkContext
 import org.apache.spark.sql.SQLContext
 
 /**
  * TPC-DS benchmark's dataset.
+ *
  * @param sqlContext An existing SQLContext.
  */
-class TPCDS
-  extends Benchmark
+class TPCDS(@transient sqlContext: SQLContext)
+  extends Benchmark(sqlContext)
   with ImpalaKitQueries
   with SimpleQueries
   with Tpcds_1_4_Queries
   with Serializable {
 
+  def this() = this(SQLContext.getOrCreate(SparkContext.getOrCreate()))
+
   /*
   def setupBroadcast(skipTables: Seq[String] = Seq("store_sales", "customer")) = {
     val skipExpr = skipTables.map(t => !('tableName === t)).reduceLeft[Column](_ && _)