I thought about doing this with a MPI subcommunicator but then it would give the wrong process ranks in the output.