diff --git a/tests/structs/test_autoscaler.py b/tests/structs/test_autoscaler.py
index cfb9560e..92d013b7 100644
--- a/tests/structs/test_autoscaler.py
+++ b/tests/structs/test_autoscaler.py
@@ -1,7 +1,7 @@
 import os
 
 from dotenv import load_dotenv
-from pytest import patch
+from unittest.mock import patch
 
 from swarms.models import OpenAIChat
 from swarms.structs import Agent