added extra testcase for mpi + fixed it