diff --git a/ufl/algorithms/apply_derivatives.py b/ufl/algorithms/apply_derivatives.py index bd9624f12..da7b61da1 100644 --- a/ufl/algorithms/apply_derivatives.py +++ b/ufl/algorithms/apply_derivatives.py @@ -30,6 +30,8 @@ Imag, Indexed, IndexSum, + Jacobian, + JacobianDeterminant, JacobianInverse, ListTensor, Product, @@ -672,7 +674,7 @@ def reference_grad(self, o): f = o.ufl_operands[0] valid_operand = f._ufl_is_in_reference_frame_ or isinstance( - f, (JacobianInverse, SpatialCoordinate) + f, (JacobianInverse, SpatialCoordinate, Jacobian, JacobianDeterminant) ) if not valid_operand: raise ValueError("ReferenceGrad can only wrap a reference frame type!")