diff --git a/tests/test_config.py b/tests/test_config.py
index dba854db..2b52f8cc 100644
--- a/tests/test_config.py
+++ b/tests/test_config.py
@@ -3,10 +3,10 @@ import os
 import openai
 import pytest
 
-from agents import set_default_openai_api, set_default_openai_client, set_default_openai_key
-from agents.models.openai_chatcompletions import OpenAIChatCompletionsModel
-from agents.models.openai_provider import OpenAIProvider
-from agents.models.openai_responses import OpenAIResponsesModel
+from cai.sdk.agents import set_default_openai_api, set_default_openai_client, set_default_openai_key
+from cai.sdk.agents.models.openai_chatcompletions import OpenAIChatCompletionsModel
+from cai.sdk.agents.models.openai_provider import OpenAIProvider
+from cai.sdk.agents.models.openai_responses import OpenAIResponsesModel
 
 
 def test_cc_no_default_key_errors(monkeypatch):
diff --git a/tests/test_doc_parsing.py b/tests/test_doc_parsing.py
index 6c7a95db..4fc13ee3 100644
--- a/tests/test_doc_parsing.py
+++ b/tests/test_doc_parsing.py
@@ -1,4 +1,4 @@
-from agents.function_schema import generate_func_documentation
+from cai.sdk.agents.function_schema import generate_func_documentation
 
 
 def func_foo_google(a: int, b: float) -> str:
diff --git a/tests/test_extension_filters.py b/tests/test_extension_filters.py
index 4cb017aa..bb70ce67 100644
--- a/tests/test_extension_filters.py
+++ b/tests/test_extension_filters.py
@@ -1,8 +1,8 @@
 from openai.types.responses import ResponseOutputMessage, ResponseOutputText
 
-from agents import Agent, HandoffInputData
-from agents.extensions.handoff_filters import remove_all_tools
-from agents.items import (
+from cai.sdk.agents import Agent, HandoffInputData
+from cai.sdk.agents.extensions.handoff_filters import remove_all_tools
+from cai.sdk.agents.items import (
     HandoffOutputItem,
     MessageOutputItem,
     ToolCallOutputItem,
diff --git a/tests/test_function_tool.py b/tests/test_function_tool.py
index 0a57aea8..a824ba32 100644
--- a/tests/test_function_tool.py
+++ b/tests/test_function_tool.py
@@ -5,8 +5,8 @@ import pytest
 from pydantic import BaseModel
 from typing_extensions import TypedDict
 
-from agents import FunctionTool, ModelBehaviorError, RunContextWrapper, function_tool
-from agents.tool import default_tool_error_function
+from cai.sdk.agents import FunctionTool, ModelBehaviorError, RunContextWrapper, function_tool
+from cai.sdk.agents.tool import default_tool_error_function
 
 
 def argless_function() -> str:
diff --git a/tests/test_function_tool_decorator.py b/tests/test_function_tool_decorator.py
index 903dd123..a62a2db2 100644
--- a/tests/test_function_tool_decorator.py
+++ b/tests/test_function_tool_decorator.py
@@ -4,8 +4,8 @@ from typing import Any, Optional
 
 import pytest
 
-from agents import function_tool
-from agents.run_context import RunContextWrapper
+from cai.sdk.agents import function_tool
+from cai.sdk.agents.run_context import RunContextWrapper
 
 
 class DummyContext: