fix mpi bug