Fix exception in SparkSinkSuite