diff --git a/tests/agents/test_agent_logging.py b/tests/agents/test_agent_logging.py
index 1d63fd4d..1439935e 100644
--- a/tests/agents/test_agent_logging.py
+++ b/tests/agents/test_agent_logging.py
@@ -1,14 +1,11 @@
-"""
-Unit tests for the Agent logging functionality.
-"""
-
 from unittest.mock import MagicMock
 import unittest
 from swarms.structs.agent import Agent
+from swarms.tools.tool_parse_exec import parse_and_execute_json
 
 # Mock parse_and_execute_json for testing
-mock_parse_and_execute_json = MagicMock()
-mock_parse_and_execute_json.return_value = {
+parse_and_execute_json = MagicMock()
+parse_and_execute_json.return_value = {
     "tool_name": "calculator",
     "args": {"numbers": [2, 2]},
     "output": "4",