diff --git a/tests/test_lookupFunction.py b/tests/test_lookupFunction.py
index 46e2882..55c62ec 100644
--- a/tests/test_lookupFunction.py
+++ b/tests/test_lookupFunction.py
@@ -5,6 +5,7 @@
 import lsst.pipe.base as pipeBase
 import lsst.utils.tests
 
+from lsst.pipe.base.all_dimensions_quantum_graph_builder import AllDimensionsQuantumGraphBuilder
 from lsst.utils import getPackageDir
 
 
@@ -73,8 +74,10 @@ def testPrerequisiteLookupFunction(self):
         pipeline = pipeBase.Pipeline("Test LookupFunction Pipeline")
         pipeline.addTask(LookupTestPipelineTask, "test")
 
-        graphBuilder = pipeBase.GraphBuilder(butler.registry)
-        graph = graphBuilder.makeGraph(pipeline, ["HSC/calib"], "output/run", None)
+        graphBuilder = AllDimensionsQuantumGraphBuilder(
+            pipeline.to_graph(), butler, input_collections=["HSC/calib"], output_run="output/run"
+        )
+        graph = graphBuilder.build()
         outputs = list(graph)
         # verify the graph contains no datasetRefs for brighter fatter kernels
         # instead of the datasetRefs that exist in the registry.