Fixed bug in PartitionAwareUnionRDD