fix partitioner to use new partitions type