diff --git a/python/dune/perftool/pdelab/driver/__init__.py b/python/dune/perftool/pdelab/driver/__init__.py
index b26af2359153546509d33abe683a28ab32b38f9f..ab34d8d8053ec47a41397c11573bc71cc59c440a 100644
--- a/python/dune/perftool/pdelab/driver/__init__.py
+++ b/python/dune/perftool/pdelab/driver/__init__.py
@@ -240,8 +240,7 @@ def generate_driver():
     if get_option("opcounter") or get_option("performance_measuring"):
         if get_option("performance_measuring"):
             assert(not get_option("opcounter"))
-        assert(any(_driver_data['form'].ufl_cell().cellname() in x for x in
-                   ["vertex", "interval", "quadrilateral", "hexahedron"]))
+        assert(isQuadrilateral(get_cell()))
         # In case of operator counting we only assemble the matrix and evaluate the residual
         # assemble_matrix_timer()
         from dune.perftool.pdelab.driver.timings import apply_jacobian_timer, evaluate_residual_timer