diff --git a/api/tests/unit_tests/core/workflow/test_runtime_state.py b/api/tests/unit_tests/core/workflow/test_runtime_state.py
index 3f3fa4195e..fe576da40a 100644
--- a/api/tests/unit_tests/core/workflow/test_runtime_state.py
+++ b/api/tests/unit_tests/core/workflow/test_runtime_state.py
@@ -157,9 +157,7 @@ class TestBindGraphRuntimeStateToGraph:
         monkeypatch.setattr(
             "core.workflow.runtime_state.ResponseStreamCoordinator",
             lambda *, variable_pool, graph: (
-                coordinator
-                if variable_pool is runtime_state.variable_pool and graph is graph
-                else None
+                coordinator if variable_pool is runtime_state.variable_pool and graph is graph else None
             ),
         )