diff --git a/python/dune/perftool/pdelab/spaces.py b/python/dune/perftool/pdelab/spaces.py
index aa3f7c21e27e50399c190279788cfdd4b3817a31..c9883dd67763fdccd37a25859bdf1804423ce6d1 100644
--- a/python/dune/perftool/pdelab/spaces.py
+++ b/python/dune/perftool/pdelab/spaces.py
@@ -71,8 +71,7 @@ def define_lfs(name, father, child):
 
 
 @preamble
-def define_lfs_size(element, restriction):
-    lfs = name_leaf_lfs(element, restriction)
+def define_lfs_size(lfs, element, restriction):
     name = name_lfs_bound(lfs)
     return "auto {} = {}.size();".format(name, lfs)
 
@@ -86,7 +85,7 @@ def define_gfs(name, father, child):
 def _name_lfs(element, restriction, tp, name):
     if len(tp) == 0:
         name_leaf_lfs(element, restriction, val=name)
-        define_lfs_size(element, restriction)
+        define_lfs_size(name, element, restriction)
         return name
 
     childname = "{}_{}".format(name, tp[0])