diff --git a/example.py b/example.py
index 1a83d232..bebdb11a 100644
--- a/example.py
+++ b/example.py
@@ -11,7 +11,4 @@ agent = Agent(
 )
 
 # Run the workflow on a task
-agent(
-    "Find a chick fil a equivalent in hayes valley"
-)
-
+agent("Find a chick fil a equivalent in hayes valley")
diff --git a/pyproject.toml b/pyproject.toml
index 3247cc64..3d3931d0 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -6,7 +6,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "swarms"
-version = "4.1.7"
+version = "4.1.8"
 description = "Swarms - Pytorch"
 license = "MIT"
 authors = ["Kye Gomez <kye@apac.ai>"]
diff --git a/swarms/models/test_fire_function.py b/swarms/models/test_fire_function.py
index e28d64ee..b6a67c37 100644
--- a/swarms/models/test_fire_function.py
+++ b/swarms/models/test_fire_function.py
@@ -1,13 +1,13 @@
-import pytest
 from unittest.mock import MagicMock
 from swarms.models.fire_function import FireFunctionCaller
 
+
 def test_fire_function_caller_run(mocker):
     # Create mock model and tokenizer
     model = MagicMock()
     tokenizer = MagicMock()
-    mocker.patch.object(FireFunctionCaller, 'model', model)
-    mocker.patch.object(FireFunctionCaller, 'tokenizer', tokenizer)
+    mocker.patch.object(FireFunctionCaller, "model", model)
+    mocker.patch.object(FireFunctionCaller, "tokenizer", tokenizer)
 
     # Create mock task and arguments
     task = "Add 2 and 3"
@@ -38,4 +38,6 @@ def test_fire_function_caller_run(mocker):
     tokenizer.batch_decode.assert_called_once_with(generated_ids)
 
     # Assert the decoded output is printed
-    assert decoded_output in mocker.patch.object(print, 'call_args_list')
+    assert decoded_output in mocker.patch.object(
+        print, "call_args_list"
+    )