diff --git a/tests/base_tool_test.py b/tests/base_tool_test.py
index e6f4f127d..dbb7fe20e 100644
--- a/tests/base_tool_test.py
+++ b/tests/base_tool_test.py
@@ -16,7 +16,7 @@ def test_creating_a_tool_using_annotation():
         my_tool.description
         == "Tool Name: Name of my tool\nTool Arguments: {'question': {'description': None, 'type': 'str'}}\nTool Description: Clear description for what this tool is useful for, you agent will need this information to use it."
     )
-    assert my_tool.args_schema.schema()["properties"] == {
+    assert my_tool.args_schema.model_json_schema()["properties"] == {
         "question": {"title": "Question", "type": "string"}
     }
     assert (
@@ -30,7 +30,7 @@ def test_creating_a_tool_using_annotation():
         converted_tool.description
         == "Tool Name: Name of my tool\nTool Arguments: {'question': {'description': None, 'type': 'str'}}\nTool Description: Clear description for what this tool is useful for, you agent will need this information to use it."
     )
-    assert converted_tool.args_schema.schema()["properties"] == {
+    assert converted_tool.args_schema.model_json_schema()["properties"] == {
         "question": {"title": "Question", "type": "string"}
     }
     assert (
@@ -54,7 +54,7 @@ def test_creating_a_tool_using_baseclass():
         my_tool.description
         == "Tool Name: Name of my tool\nTool Arguments: {'question': {'description': None, 'type': 'str'}}\nTool Description: Clear description for what this tool is useful for, you agent will need this information to use it."
     )
-    assert my_tool.args_schema.schema()["properties"] == {
+    assert my_tool.args_schema.model_json_schema()["properties"] == {
         "question": {"title": "Question", "type": "string"}
     }
     assert (
@@ -68,7 +68,7 @@ def test_creating_a_tool_using_baseclass():
         converted_tool.description
         == "Tool Name: Name of my tool\nTool Arguments: {'question': {'description': None, 'type': 'str'}}\nTool Description: Clear description for what this tool is useful for, you agent will need this information to use it."
     )
-    assert converted_tool.args_schema.schema()["properties"] == {
+    assert converted_tool.args_schema.model_json_schema()["properties"] == {
         "question": {"title": "Question", "type": "string"}
     }
     assert (