diff --git a/tests/test_e2e_v27.py b/tests/test_e2e_v27.py
index ef1a4e0..9d3ebb9 100644
--- a/tests/test_e2e_v27.py
+++ b/tests/test_e2e_v27.py
@@ -14,6 +14,7 @@ from pathlib import Path
 from typing import Any, Dict, List, Optional
 
 import pytest
+from unittest.mock import MagicMock
 from fastapi.testclient import TestClient
 
 # 指向部署目录