Fix MPI crash