diff --git a/sec-gemini-python/tests/__init__.py b/sec-gemini-python/tests/__init__.py
new file mode 100644
index 0000000..8b13789
--- /dev/null
+++ b/sec-gemini-python/tests/__init__.py
@@ -0,0 +1 @@
+
diff --git a/sec-gemini-python/tests/conftest.py b/sec-gemini-python/tests/conftest.py
index bb1200a..f4f262d 100644
--- a/sec-gemini-python/tests/conftest.py
+++ b/sec-gemini-python/tests/conftest.py
@@ -19,7 +19,8 @@
 import httpx
 import pytest
 from pytest_httpx import HTTPXMock
-from utils import require_env_variable
+
+from .utils import require_env_variable
 
 from sec_gemini import SecGemini
 from sec_gemini.models.enums import State, UserType
diff --git a/sec-gemini-python/tests/test_local_tools_integration.py b/sec-gemini-python/tests/test_local_tools_integration.py
index 6d52c16..39ab6ff 100644
--- a/sec-gemini-python/tests/test_local_tools_integration.py
+++ b/sec-gemini-python/tests/test_local_tools_integration.py
@@ -9,6 +9,8 @@
 from mcp.server.fastmcp.tools import Tool
 from rich.markdown import Markdown
 
+from .utils import async_require_env_variable
+
 from sec_gemini import SecGemini, State
 from sec_gemini.models.enums import MessageType, MimeType
 from sec_gemini.models.local_tool import LocalTool
@@ -270,6 +272,7 @@ def tool_with_optional_int(value: int | None = None):
     print()
 
 
+@async_require_env_variable("SEC_GEMINI_API_KEY")
 @pytest.mark.asyncio
 async def test_list_dir_and_sha256_local_tools() -> None:
   """Tests the local tool integration by streaming a prompt that requires file listing and hashing."""
diff --git a/sec-gemini-python/tests/test_secgemini.py b/sec-gemini-python/tests/test_secgemini.py
index 69ef288..533aebc 100644
--- a/sec-gemini-python/tests/test_secgemini.py
+++ b/sec-gemini-python/tests/test_secgemini.py
@@ -22,10 +22,11 @@
 from typing import TypeVar
 
 import pytest
-from conftest import MOCK_SEC_GEMINI_API_HOST
 from pytest_httpx import HTTPXMock
 from rich import print as rprint
-from utils import (
+
+from .conftest import MOCK_SEC_GEMINI_API_HOST
+from .utils import (
   async_require_env_variable,
   parse_secgemini_response,
   require_env_variable,