Save a few lines of code

This might even save a few seconds on big runs when using PETSc