Adapt coloring for MPI