diff --git a/tests/test_agent.py b/tests/test_agent.py
index d8eb119..4eb3e70 100644
--- a/tests/test_agent.py
+++ b/tests/test_agent.py
@@ -1,6 +1,5 @@
 import pytest
-from livekit.agents import AgentSession, llm
-from livekit.agents.voice.run_result import mock_tools
+from livekit.agents import AgentSession, llm, mock_tools
 from livekit.plugins import openai
 
 from agent import Assistant