Forgot to pre-allocate DistributedState.donePartitions