diff --git a/agentic_security/mcp/client.py b/agentic_security/mcp/client.py
index 0509827..79c3758 100644
--- a/agentic_security/mcp/client.py
+++ b/agentic_security/mcp/client.py
@@ -1,7 +1,8 @@
-from mcp import ClientSession, StdioServerParameters, types
-from mcp.client.stdio import stdio_client
 import asyncio
 
+from mcp import ClientSession, StdioServerParameters
+from mcp.client.stdio import stdio_client
+
 # Create server parameters for stdio connection
 server_params = StdioServerParameters(
     command="python",  # Executable
diff --git a/agentic_security/mcp/main.py b/agentic_security/mcp/main.py
index 45e1202..0e754dd 100644
--- a/agentic_security/mcp/main.py
+++ b/agentic_security/mcp/main.py
@@ -1,6 +1,3 @@
-import asyncio
-import os
-
 import httpx
 from mcp.server.fastmcp import FastMCP