diff --git a/libs/partners/ollama/tests/unit_tests/test_chat_models_gpt_oss.py b/libs/partners/ollama/tests/unit_tests/test_chat_models_gpt_oss.py
index 40eb20bfa5a..31f96e50e5d 100644
--- a/libs/partners/ollama/tests/unit_tests/test_chat_models_gpt_oss.py
+++ b/libs/partners/ollama/tests/unit_tests/test_chat_models_gpt_oss.py
@@ -321,7 +321,7 @@ class TestGptOssIntegration:
         for tool_def in tools:
             assert tool_def["type"] == "function"
             assert "function" in tool_def
-            func = tool["function"]
+            func = tool_def["function"]
             assert "name" in func
             assert "description" in func
             assert "parameters" in func
@@ -397,3 +397,4 @@ class TestChatParamsWithGptOss:
 
 
 
+