Fix MPI tests