Fixed mpi type issue