diff --git a/test/agentchat/contrib/test_reasoning_agent.py b/test/agentchat/contrib/test_reasoning_agent.py
index 19b96cad45..dbca346360 100644
--- a/test/agentchat/contrib/test_reasoning_agent.py
+++ b/test/agentchat/contrib/test_reasoning_agent.py
@@ -15,6 +15,7 @@
 
 import pytest
 
+import autogen
 from autogen.agentchat.contrib.reasoning_agent import ReasoningAgent, ThinkNode, visualize_tree
 from autogen.agentchat.user_proxy_agent import UserProxyAgent