Fixing broken test due to change in zipPartitions