Update test_mpi.cpp