diff --git a/benchmarks/pure_graph_bench.py b/benchmarks/pure_graph_bench.py
index 83f063e96..de376735e 100644
--- a/benchmarks/pure_graph_bench.py
+++ b/benchmarks/pure_graph_bench.py
@@ -28,9 +28,7 @@ class PureGraphBench:
         for i in range(node_count):
             node_name = f"node_{i}"
             self.nodes[node_name] = NodeConfig(
-                name=node_name,
-                type=NodeType.FUNCTION,
-                function=f"fn_{i}"
+                name=node_name, type=NodeType.FUNCTION, function=f"fn_{i}"
             )
             self.edges.append(Edge(source=previous, target=node_name))
             previous = node_name
@@ -43,9 +41,7 @@ class PureGraphBench:
         }
 
         self.graph = PureGraph(
-            name=f"bench_graph_{node_count}",
-            nodes=self.nodes,
-            edges=self.edges
+            name=f"bench_graph_{node_count}", nodes=self.nodes, edges=self.edges
         )
 
     def time_topological_order(self, node_count: int) -> None: