diff --git a/docs/concepts/state-machine.rst b/docs/concepts/state-machine.rst
index 2fea48aea..1e3ce10d7 100644
--- a/docs/concepts/state-machine.rst
+++ b/docs/concepts/state-machine.rst
@@ -171,7 +171,7 @@ in a web-server (create a graph once, application many times), import the graph
 
 .. code-block:: python
 
-    from burr.core import ApplicationBuilder, default, expr
+    from burr.core import ApplicationBuilder, GraphBuilder
     graph = (
         GraphBuilder()
         .with_actions(human_input, ai_response)
@@ -183,7 +183,7 @@ in a web-server (create a graph once, application many times), import the graph
     app = (
         ApplicationBuilder()
         .with_graph(graph)
-        with_state(chat_history=[])
+        .with_state(chat_history=[])
         .with_entrypoint("human_input")
         .build()
     )