another one

firedrakeproject · Nov 23, 2023 · 068f6c3 · 068f6c3
1 parent 0499992
commit 068f6c3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ufl/algorithms/change_to_reference.py b/ufl/algorithms/change_to_reference.py
@@ -13,6 +13,7 @@
 from ufl.core.multiindex import indices
 from ufl.corealg.map_dag import map_expr_dag
 from ufl.corealg.multifunction import MultiFunction
+from ufl.domain import extract_unique_domain
 from ufl.tensors import as_tensor
 
 """
@@ -139,7 +140,7 @@ def grad(self, o):
             f = ReferenceValue(f)
 
         # Get domain and create Jacobian inverse object
-        domain = o.ufl_domain()
+        domain = extract_unique_domain(o)
         Jinv = JacobianInverse(domain)
 
         if is_cellwise_constant(Jinv):