MPI communication fixed