Fixed MPI failure

Old code was going over all blocks, should just go over local blocks