diff --git a/python/dune/codegen/pdelab/tensors.py b/python/dune/codegen/pdelab/tensors.py
index 7a86ba52eff5f255c3d040d31951043a65266890..a277e751b4a2e878b3248ca582b219633f46d874 100644
--- a/python/dune/codegen/pdelab/tensors.py
+++ b/python/dune/codegen/pdelab/tensors.py
@@ -177,7 +177,7 @@ def pymbolic_matrix_inverse(o, visitor):
     expr = o.ufl_operands[0]
     name = name_assembled_tensor(expr, visitor)
 
-    if expr.shape[0] <= 3:
+    if expr.ufl_shape[0] <= 3:
         name = name_matrix_inverse(name, expr.ufl_shape, visitor)
     else:
         instruction(code="{}.invert();".format(name),