diff --git a/tests/test_agent_config.py b/tests/test_agent_config.py
index 44339dad..19f97cdf 100644
--- a/tests/test_agent_config.py
+++ b/tests/test_agent_config.py
@@ -1,7 +1,7 @@
 import pytest
 from pydantic import BaseModel
 
-from agents import Agent, Handoff, RunContextWrapper, Runner, handoff
+from cai.sdk.agents import Agent, Handoff, RunContextWrapper, Runner, handoff
 
 
 @pytest.mark.asyncio