diff --git a/tests/test_models.py b/tests/test_models.py
index 2f6a7da..903329d 100644
--- a/tests/test_models.py
+++ b/tests/test_models.py
@@ -4,21 +4,20 @@ import pytest
 from pydantic import ValidationError
 
 from mcp_server_cli.models import (
-    MCPRequest,
-    MCPResponse,
-    MCPNotification,
-    MCPMethod,
-    ToolDefinition,
-    ToolSchema,
-    ToolParameter,
-    ToolCallParams,
-    ToolCallResult,
+    AppConfig,
     InitializeParams,
     InitializeResult,
-    ServerConfig,
     LocalLLMConfig,
+    MCPMethod,
+    MCPRequest,
+    MCPResponse,
     SecurityConfig,
-    AppConfig,
+    ServerConfig,
+    ToolCallParams,
+    ToolCallResult,
+    ToolDefinition,
+    ToolParameter,
+    ToolSchema,
 )