diff --git a/gusto/equations.py b/gusto/equations.py index a1421d0ea..c01a0ef38 100644 --- a/gusto/equations.py +++ b/gusto/equations.py @@ -985,7 +985,8 @@ def __init__(self, domain, parameters, Omega=None, sponge=None, # -------------------------------------------------------------------- # # Gravitational Term # -------------------------------------------------------------------- # - gravity_form = subject(prognostic(Term(g*inner(domain.k, w)*dx), 'u'), self.X) + gravity_form = name_label(subject(prognostic(Term(g*inner(domain.k, w)*dx), + 'u'), self.X), "gravity") residual = (mass_form + adv_form + pressure_gradient_form + gravity_form)