diff --git a/.gitignore b/.gitignore
index 98167594..a336e116 100644
--- a/.gitignore
+++ b/.gitignore
@@ -29,6 +29,10 @@ error.txt
 # C extensions
 *.so
 
+
+errors.txt
+
+
 # Distribution / packaging
 .Python
 build/
diff --git a/tests/structs/flow.py b/tests/structs/flow.py
index 0f20e392..b9f251dc 100644
--- a/tests/structs/flow.py
+++ b/tests/structs/flow.py
@@ -1,4 +1,7 @@
 import pytest
+import json
+from unittest.mock import MagicMock
+
 from unittest.mock import patch
 import os
 from swarms.structs.flow import Flow, stop_when_repeats