Updated ShuffledRDD based in its API change in Spark.