diff --git a/docs/docs/concepts/functional_api.md b/docs/docs/concepts/functional_api.md
index 1a44f249b..d6a3ebb1f 100644
--- a/docs/docs/concepts/functional_api.md
+++ b/docs/docs/concepts/functional_api.md
@@ -23,9 +23,11 @@ This provides a minimal abstraction for building workflows with state management
 Below we demonstrate a simple application that writes an essay and [interrupts](human_in_the_loop.md) to request human review.
 
 ```python
+from langgraph.checkpoint.memory import MemorySaver
 from langgraph.func import entrypoint, task
 from langgraph.types import interrupt
 
+
 @task
 def write_essay(topic: str) -> str:
     """Write an essay about the given topic."""