shouldn't the 4 here be `numPartitionPerMachine`?