diff --git a/tests/agents/test_agent_logging.py b/tests/agents/test_agent_logging.py
index 1439935e..1d63fd4d 100644
--- a/tests/agents/test_agent_logging.py
+++ b/tests/agents/test_agent_logging.py
@@ -1,11 +1,14 @@
+"""
+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
-parse_and_execute_json = MagicMock()
-parse_and_execute_json.return_value = {
+mock_parse_and_execute_json = MagicMock()
+mock_parse_and_execute_json.return_value = {
     "tool_name": "calculator",
     "args": {"numbers": [2, 2]},
     "output": "4",