fix broken two-rdd pileup flatmap unit test