[firedrake] close to fixing my nonlinear QG solver