diff --git a/langgraph/graph/__init__.py b/langgraph/graph/__init__.py
index 1173035b5..8fac44cfa 100644
--- a/langgraph/graph/__init__.py
+++ b/langgraph/graph/__init__.py
@@ -1,4 +1,5 @@
 from langgraph.graph.graph import END, Graph
+from langgraph.graph.message import MessageGraph
 from langgraph.graph.state import StateGraph
 
-__all__ = ["END", "Graph", "StateGraph"]
+__all__ = ["END", "Graph", "StateGraph", "MessageGraph"]