diff --git a/tests/integration/test_py.py b/tests/integration/test_py.py
index ce6accf9..3baa7592 100644
--- a/tests/integration/test_py.py
+++ b/tests/integration/test_py.py
@@ -4,8 +4,8 @@
 
 from guppylang.decorator import guppy
 from guppylang.module import GuppyModule
+from guppylang.prelude.builtins import py
 from guppylang.prelude.quantum import qubit, quantum
-from tests.integration.util import py
 from tests.util import compile_guppy
 
 tket2_installed = find_spec("tket2") is not None
diff --git a/tests/integration/util.py b/tests/integration/util.py
index 40232e58..fa3a7d4a 100644
--- a/tests/integration/util.py
+++ b/tests/integration/util.py
@@ -1,6 +1,3 @@
-from typing import Any
-
-
 class Decorator:
     def __matmul__(self, other):
         return None
@@ -9,8 +6,3 @@ def __matmul__(self, other):
 # Dummy names to import to avoid errors for `_@functional` pseudo-decorator:
 functional = Decorator()
 _ = Decorator()
-
-
-def py(*args: Any) -> Any:
-    """Dummy function to import to avoid errors for `py(...)` expressions"""
-    raise NotImplementedError