diff --git a/agentverse/agentverse.py b/agentverse/agentverse.py
index 8b1505075..62c815596 100644
--- a/agentverse/agentverse.py
+++ b/agentverse/agentverse.py
@@ -3,7 +3,7 @@
 from typing import List
 
 # from agentverse.agents import Agent
-from agentverse.agents.conversation_agent import BaseAgent
+from agentverse.agents.base import BaseAgent
 from agentverse.environments import BaseEnvironment
 from agentverse.initialization import load_agent, load_environment, prepare_task_config