diff --git a/cognee/tests/integration/run_toy_tasks/test_run_tasks.py b/cognee/tests/integration/run_toy_tasks/test_run_tasks.py
index 2cc0467ba..d0a2af80b 100644
--- a/cognee/tests/integration/run_toy_tasks/test_run_tasks.py
+++ b/cognee/tests/integration/run_toy_tasks/test_run_tasks.py
@@ -33,8 +33,9 @@ async def run_and_check_tasks():
     results = [5, 7, 9, 11, 13, 15, 17, 19, 21, 23]
     index = 0
     async for result in pipeline:
-        print(result)
-        assert result == results[index]
+        assert (
+            result == results[index]
+        ), f"at {index = }: {result = } != {results[index] = }"
         index += 1
 
 
diff --git a/cognee/tests/integration/run_toy_tasks/test_run_tasks_from_queue.py b/cognee/tests/integration/run_toy_tasks/test_run_tasks_from_queue.py
index cf8047349..e57b16f39 100644
--- a/cognee/tests/integration/run_toy_tasks/test_run_tasks_from_queue.py
+++ b/cognee/tests/integration/run_toy_tasks/test_run_tasks_from_queue.py
@@ -31,8 +31,9 @@ async def pipeline(data_queue):
     results = [2, 4, 6, 8, 10, 12, 14, 16, 18, 20]
     index = 0
     async for result in tasks_run:
-        print(result)
-        assert result == results[index]
+        assert (
+            result == results[index]
+        ), f"at {index = }: {result = } != {results[index] = }"
         index += 1