diff --git a/tests/unit/test_tool.py b/tests/unit/test_tool.py
index 13e40fd88..cd1013f8a 100644
--- a/tests/unit/test_tool.py
+++ b/tests/unit/test_tool.py
@@ -2,7 +2,7 @@
 
 import pytest
 from unittest.mock import MagicMock, AsyncMock
-from pydantic import ValidationError
+from jsonschema.exceptions import ValidationError
 
 from astrbot.core.agent.tool import (
     FunctionTool,