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 e43cffdeb04..446e52ff1d7 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
@@ -288,7 +288,7 @@ class TestGptOssResponseParsing:
 class TestGptOssIntegration:
     """Integration tests for gpt-oss model with ChatOllama."""
     
-    @patch("ollama.Client")
+    @patch("langchain_ollama.chat_models.Client")
     def test_invoke_with_tools(self, mock_client_class: MagicMock) -> None:
         """Test invoking gpt-oss model with tools."""
         # Create mock client instance
@@ -410,3 +410,4 @@ class TestChatParamsWithGptOss:
 
 
 
+