diff --git a/python/dune/perftool/ufl/visitor.py b/python/dune/perftool/ufl/visitor.py
index 8009bc2bbd42d035291016e9e199061cde3fd55f..ccc6cb24091e6b48b53d0a86292e34e93dd42f23 100644
--- a/python/dune/perftool/ufl/visitor.py
+++ b/python/dune/perftool/ufl/visitor.py
@@ -156,7 +156,7 @@ class UFL2LoopyVisitor(ModifiedTerminalTracker):
 
             predicates = predicates.union(['{} == {}'.format(name, self.subdomain_id)])
 
-        from dune.perftool.loopy.functions import PDELabAccumulationFunction
+        from dune.perftool.pdelab.argument import PDELabAccumulationFunction
         from pymbolic.primitives import Call
         expr = Call(PDELabAccumulationFunction(accumvar, len(test_ma)), tuple(a for a in accumargs) + (pymbolic_expr,))
 
@@ -430,4 +430,4 @@ class UFL2LoopyVisitor(ModifiedTerminalTracker):
 
     def facet_jacobian_determinant(self, o):
         from dune.perftool.pdelab.geometry import name_facet_jacobian_determinant
-        return Variable(name_facet_jacobian_determinant())
\ No newline at end of file
+        return Variable(name_facet_jacobian_determinant())