[MPI] Fixed some bugs