Updated to call PartitionCollector regardless of if the partition completes successfully