fix regression in partitionWith