diff --git a/python/dune/perftool/ufl/transformations/axiparallel.py b/python/dune/perftool/ufl/transformations/axiparallel.py
index 5f5f26af2b27bd07ee1e6f788d9963d16a780e22..129304a8e2bc5dd4ea3cf6b8e1378152d96a4d34 100644
--- a/python/dune/perftool/ufl/transformations/axiparallel.py
+++ b/python/dune/perftool/ufl/transformations/axiparallel.py
@@ -125,6 +125,8 @@ class DiagonalJITTester(MultiFunction):
         p, i = o.ufl_operands
         if isinstance(p, JacobianInverse):
             assert i[0] == i[1]
+        else:
+            self.call(p)
 
 
 def test_diagonal_jacobian(e):