Exactly! If you don't to that, you would have to introduce conditional statements to deal with the communication at the edge of the domain where MPI_PROC_NULL is returned. As an exercise in debugging you could try to remove those statements and see how it fails by reducing the size of the domain to small matrices and one time step only.
Forum