diff --git a/README.md b/README.md
index fd52ed38d1faa6389cc6d9902461b579869e1fb4..9b6de4060d90f8b41e3bea580d53360c2fb62c1b 100644
--- a/README.md
+++ b/README.md
@@ -48,7 +48,7 @@ agent := fantasy.NewAgent(model, fantasy.WithTools(cuteDogTool))
 
 // Put that agent to work!
 const prompt = "Find all the cute dogs in Silver Lake, Los Angeles."
-result, err := agent.Generate(context.Background(), fantasy.AgentCall{Prompt: prompt})
+result, err := agent.Generate(ctx, fantasy.AgentCall{Prompt: prompt})
 if err != nil {
     fmt.Fprintln(os.Stderr, "Oof:", err)
     os.Exit(1)