diff --git a/lang_agent/graphs/__init__.py b/lang_agent/graphs/__init__.py
index 713160b..360dc4f 100644
--- a/lang_agent/graphs/__init__.py
+++ b/lang_agent/graphs/__init__.py
@@ -1,9 +1,9 @@
 import tyro
 
-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.react import ReactGraphConfig, ReactGraph
+from lang_agent.graphs.routing import RoutingConfig, RoutingGraph
+from lang_agent.graphs.dual_path import DualConfig, DualPathGraph
+from lang_agent.graphs.vision_routing import VisionRoutingConfig, VisionRoutingGraph
 
 graph_dict = {
     "react": ReactGraphConfig(),