diff --git a/python/dune/perftool/__init__.py b/python/dune/perftool/__init__.py
index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..fa06bca3d581b54ad724163642d813f35181d9c7 100644
--- a/python/dune/perftool/__init__.py
+++ b/python/dune/perftool/__init__.py
@@ -0,0 +1,4 @@
+class Restriction:
+    NONE = 0
+    POSITIVE = 1
+    NEGATIVE = 2
\ No newline at end of file
diff --git a/python/dune/perftool/restriction.py b/python/dune/perftool/restriction.py
deleted file mode 100644
index fa06bca3d581b54ad724163642d813f35181d9c7..0000000000000000000000000000000000000000
--- a/python/dune/perftool/restriction.py
+++ /dev/null
@@ -1,4 +0,0 @@
-class Restriction:
-    NONE = 0
-    POSITIVE = 1
-    NEGATIVE = 2
\ No newline at end of file
diff --git a/python/dune/perftool/transformer.py b/python/dune/perftool/transformer.py
index 903351ca75f064636e5384d9eaab5a30735be481..3213e572f48ab48ae77e5408be613241494df004 100644
--- a/python/dune/perftool/transformer.py
+++ b/python/dune/perftool/transformer.py
@@ -7,7 +7,7 @@ import loopy
 import numpy
 import ufl
 
-from dune.perftool.restriction import Restriction
+from dune.perftool import Restriction
 
 # Define the generators that are used here
 from dune.perftool.generation import generator_factory
diff --git a/python/dune/perftool/ufl/modified_terminals.py b/python/dune/perftool/ufl/modified_terminals.py
index b8c3900c167d5224653adf0bf8c7787154cc64cc..00eb30da83e555782faebf2257fb895b8a1dabe5 100644
--- a/python/dune/perftool/ufl/modified_terminals.py
+++ b/python/dune/perftool/ufl/modified_terminals.py
@@ -1,7 +1,7 @@
 """ A module mimicking some functionality of uflacs' modified terminals """
 
 from ufl.algorithms import MultiFunction
-from dune.perftool.restriction import Restriction
+from dune.perftool import Restriction
 
 
 class ModifiedTerminalTracker(MultiFunction):