diff --git a/cognee/tests/unit/interfaces/graph/get_graph_from_huge_model_test.py b/cognee/tests/unit/interfaces/graph/get_graph_from_huge_model_test.py
index 4a4619e19..1f7b5cbf7 100644
--- a/cognee/tests/unit/interfaces/graph/get_graph_from_huge_model_test.py
+++ b/cognee/tests/unit/interfaces/graph/get_graph_from_huge_model_test.py
@@ -73,10 +73,7 @@ async def test_circular_reference_extraction():
 
     for code_file in code_files:
         code_file.depends_on.extend(
-            [
-                code_files[random.randint(0, len(code_files) - 1)]
-                for _ in range(2)
-            ]
+            [code_files[random.randint(0, len(code_files) - 1)] for _ in range(2)]
         )
         code_file.contains.extend(
             [