[firedrake] Faster way to do this?