Make pull with COM from previous step work with MPI