diff --git a/tests/fake_model.py b/tests/fake_model.py
index ecbb7583..a5cff359 100644
--- a/tests/fake_model.py
+++ b/tests/fake_model.py
@@ -5,19 +5,19 @@ from typing import Any
 
 from openai.types.responses import Response, ResponseCompletedEvent
 
-from agents.agent_output import AgentOutputSchema
-from agents.handoffs import Handoff
-from agents.items import (
+from cai.sdk.agents.agent_output import AgentOutputSchema
+from cai.sdk.agents.handoffs import Handoff
+from cai.sdk.agents.items import (
     ModelResponse,
     TResponseInputItem,
     TResponseOutputItem,
     TResponseStreamEvent,
 )
-from agents.model_settings import ModelSettings
-from agents.models.interface import Model, ModelTracing
-from agents.tool import Tool
-from agents.tracing import SpanError, generation_span
-from agents.usage import Usage
+from cai.sdk.agents.model_settings import ModelSettings
+from cai.sdk.agents.models.interface import Model, ModelTracing
+from cai.sdk.agents.tool import Tool
+from cai.sdk.agents.tracing import SpanError, generation_span
+from cai.sdk.agents.usage import Usage
 
 
 class FakeModel(Model):
diff --git a/tests/test_agent_hooks.py b/tests/test_agent_hooks.py
index 33107cba..28caa58e 100644
--- a/tests/test_agent_hooks.py
+++ b/tests/test_agent_hooks.py
@@ -7,11 +7,11 @@ from typing import Any
 import pytest
 from typing_extensions import TypedDict
 
-from agents.agent import Agent
-from agents.lifecycle import AgentHooks
-from agents.run import Runner
-from agents.run_context import RunContextWrapper, TContext
-from agents.tool import Tool
+from cai.sdk.agents.agent import Agent
+from cai.sdk.agents.lifecycle import AgentHooks
+from cai.sdk.agents.run import Runner
+from cai.sdk.agents.run_context import RunContextWrapper, TContext
+from cai.sdk.agents.tool import Tool
 
 from .fake_model import FakeModel
 from .test_responses import (
diff --git a/tests/test_responses.py b/tests/test_responses.py
index 6b91bf8c..1b4fec47 100644
--- a/tests/test_responses.py
+++ b/tests/test_responses.py
@@ -9,7 +9,7 @@ from openai.types.responses import (
     ResponseOutputText,
 )
 
-from agents import (
+from cai.sdk.agents import (
     Agent,
     FunctionTool,
     Handoff,