Update SparkSinkSuite.scala