diff --git a/python/dune/perftool/pdelab/driver/timings.py b/python/dune/perftool/pdelab/driver/timings.py
index 4ba2d8318de885030240adabcf3e9e139bb20524..cdeb6b1d5e78e8794c3973343e0c32e76535b0d4 100644
--- a/python/dune/perftool/pdelab/driver/timings.py
+++ b/python/dune/perftool/pdelab/driver/timings.py
@@ -44,12 +44,22 @@ def dump_dof_numbers(stream):
     include_file("dune/pdelab/common/partitionviewentityset.hh", filetag="driver")
     gvt = type_leafview()
     gv = name_leafview()
-    return ["Dune::PDELab::NonOverlappingEntitySet<{}> es({});".format(gvt, gv),
-            "{} << \"{} \" << {} << \" dofs dofs \" << {}.maxLocalSize() * es.size(0) << std::endl;".format(stream,
-                                                                                                            level,
-                                                                                                            ident,
-                                                                                                            name_trial_gfs())
-            ]
+
+    from dune.perftool.pdelab.driver import get_trial_element, isDG
+    if isDG(get_trial_element()):
+        return ["Dune::PDELab::NonOverlappingEntitySet<{}> es({});".format(gvt, gv),
+                "{} << \"{} \" << {} << \" dofs dofs \" << {}.maxLocalSize() * es.size(0) << std::endl;".format(stream,
+                                                                                                                level,
+                                                                                                                ident,
+                                                                                                                name_trial_gfs())
+                ]
+    else:
+        return ["{} << \"{} \" << {} << \" dofs dofs \" << {}.() << std::endl;".format(stream,
+                                                                                       level,
+                                                                                       ident,
+                                                                                       name_trial_gfs())
+                ]
+
 
 
 @preamble(section="timings")