Forgot to use PartitionSet.values() instead of its iterator