diff --git a/tests/inference/test_inference.py b/tests/inference/test_inference.py
index 20dd3c235..7828f0009 100644
--- a/tests/inference/test_inference.py
+++ b/tests/inference/test_inference.py
@@ -175,12 +175,11 @@ class TestInference:
     #
     # Returns a "_client_graph", which is client-graph pair corresponding to an initialized server
     # The "graph" is the default graph
-    @fixture(scope="class", params=comfy_graph_list, ids=comfy_graph_ids, autouse=False)
-    def _client_graph(self, request, args_pytest, comfy_background_server) -> (ComfyClient, ComfyGraph):
+    @fixture(scope="function", params=comfy_graph_list, ids=comfy_graph_ids, autouse=False)
+    def _client_graph(self, request, comfy_background_server) -> (ComfyClient, ComfyGraph):
         comfy_graph = request.param
-
         # Start client
-        comfy_client = self.start_client(args_pytest["listen"], args_pytest["port"])
+        comfy_client = self.start_client(comfy_background_server.listen, comfy_background_server.port)
 
         # Warm up pipeline
         comfy_client.get_images(graph=comfy_graph.graph, save=False)