diff --git a/libs/genai/tests/unit_tests/test_chat_models.py b/libs/genai/tests/unit_tests/test_chat_models.py
index 1af737ef..51641e25 100644
--- a/libs/genai/tests/unit_tests/test_chat_models.py
+++ b/libs/genai/tests/unit_tests/test_chat_models.py
@@ -547,7 +547,7 @@ def test_parse_response_candidate(raw_candidate: Dict, expected: AIMessage) -> N
 
 
 def test_serialize() -> None:
-    llm = ChatGoogleGenerativeAI(model="gemini-pro-1.5", google_api_key="test-key")
+    llm = ChatGoogleGenerativeAI(model="gemini-pro-1.5", api_key=SecretStr("test-key"))
     serialized = dumps(llm)
     llm_loaded = loads(
         serialized,
diff --git a/libs/genai/tests/unit_tests/test_genai_aqa.py b/libs/genai/tests/unit_tests/test_genai_aqa.py
index c8cd521b..4deaaf9d 100644
--- a/libs/genai/tests/unit_tests/test_genai_aqa.py
+++ b/libs/genai/tests/unit_tests/test_genai_aqa.py
@@ -3,11 +3,9 @@
 import google.ai.generativelanguage as genai
 import pytest
 
-from langchain_google_genai import (
-    AqaInput,
-    GenAIAqa,
-)
+from langchain_google_genai import AqaInput, GenAIAqa
 from langchain_google_genai import _genai_extension as genaix
+from langchain_google_genai._enums import HarmBlockThreshold, HarmCategory
 
 # Make sure the tests do not hit actual production servers.
 genaix.set_config(
@@ -54,8 +52,8 @@ def test_invoke(mock_generate_answer: MagicMock) -> None:
         answer_style=genai.GenerateAnswerRequest.AnswerStyle.EXTRACTIVE,
         safety_settings=[
             genai.SafetySetting(
-                category=genai.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
-                threshold=genai.SafetySetting.HarmBlockThreshold.BLOCK_LOW_AND_ABOVE,
+                category=HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
+                threshold=HarmBlockThreshold.BLOCK_LOW_AND_ABOVE,
             )
         ],
     )
@@ -80,11 +78,10 @@ def test_invoke(mock_generate_answer: MagicMock) -> None:
     assert len(request.safety_settings) == 1
     assert (
         request.safety_settings[0].category
-        == genai.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT
+        == HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT
     )
     assert (
-        request.safety_settings[0].threshold
-        == genai.SafetySetting.HarmBlockThreshold.BLOCK_LOW_AND_ABOVE
+        request.safety_settings[0].threshold == HarmBlockThreshold.BLOCK_LOW_AND_ABOVE
     )
 
     assert request.temperature == 0.5
diff --git a/libs/genai/tests/unit_tests/test_llms.py b/libs/genai/tests/unit_tests/test_llms.py
index 29a68980..64892145 100644
--- a/libs/genai/tests/unit_tests/test_llms.py
+++ b/libs/genai/tests/unit_tests/test_llms.py
@@ -1,3 +1,5 @@
+from langchain_core.pydantic_v1 import SecretStr
+
 from langchain_google_genai.llms import GoogleGenerativeAI, GoogleModelFamily
 
 
@@ -10,7 +12,7 @@ def test_model_family() -> None:
 
 def test_tracing_params() -> None:
     # Test standard tracing params
-    llm = GoogleGenerativeAI(model="gemini-pro", google_api_key="foo")  # type: ignore[call-arg]
+    llm = GoogleGenerativeAI(model="gemini-pro", google_api_key=SecretStr("foo"))
     ls_params = llm._get_ls_params()
     assert ls_params == {
         "ls_provider": "google_genai",
@@ -23,7 +25,7 @@ def test_tracing_params() -> None:
         model="gemini-pro",
         temperature=0.1,
         max_output_tokens=10,
-        google_api_key="foo",  # type: ignore[call-arg]
+        google_api_key=SecretStr("foo"),
     )
     ls_params = llm._get_ls_params()
     assert ls_params == {