diff --git a/lang_agent/graphs/__init__.py b/lang_agent/graphs/__init__.py
index de9cc71..713160b 100644
--- a/lang_agent/graphs/__init__.py
+++ b/lang_agent/graphs/__init__.py
@@ -4,14 +4,12 @@ from lang_agent.graphs.react import ReactGraphConfig
 from lang_agent.graphs.routing import RoutingConfig
 from lang_agent.graphs.dual_path import DualConfig
 from lang_agent.graphs.vision_routing import VisionRoutingConfig
-# from lang_agent.graphs.xiaoai_demo import XiaoAiConfig
 
 graph_dict = {
     "react": ReactGraphConfig(),
     "route": RoutingConfig(),
     "dual": DualConfig(),
-    "vision": VisionRoutingConfig(),
-    # "ai_demo": XiaoAiConfig()
+    "vision": VisionRoutingConfig()
 }
 
 graph_union = tyro.extras.subcommand_type_from_defaults(graph_dict, prefix_names=False)