Ah, indeed. That should work with `ufl.replace` though I think.