Changed PartitionPruningRDD's split to make sure it returns the correct split index.