diff --git a/tests/util/get_passed_env.py b/tests/util/get_passed_env.py
index d9c4ba6..54a3280 100644
--- a/tests/util/get_passed_env.py
+++ b/tests/util/get_passed_env.py
@@ -1,6 +1,6 @@
 import os
 
-from bonobo import Graph
+import bonobo
 
 
 def extract():
@@ -14,9 +14,7 @@ def load(s: str):
     print(s)
 
 
-graph = Graph(extract, load)
+graph = bonobo.Graph(extract, load)
 
 if __name__ == '__main__':
-    from bonobo import run
-
-    run(graph)
+    bonobo.run(graph)