diff --git a/src/content/docs/docs/flows.mdx b/src/content/docs/docs/flows.mdx
index 79ec7bcf..62e0e513 100644
--- a/src/content/docs/docs/flows.mdx
+++ b/src/content/docs/docs/flows.mdx
@@ -96,6 +96,7 @@ func main() {
 
 	g := genkit.Init(ctx,
 		genkit.WithPlugins(&googlegenai.GoogleAI{}),
+		genkit.WithDefaultModel("googleai/gemini-2.5-flash"),
 	)
 
 	type MenuSuggestionInput struct {
@@ -245,7 +246,7 @@ type MenuItem struct {
 }
 
 menuSuggestionFlow := genkit.DefineFlow(g, "menuSuggestionFlow",
-	func(ctx context.Context, input MenuSuggestionInput) (MenuItem, error) {
+	func(ctx context.Context, input MenuSuggestionInput) (*MenuItem, error) {
 		item, _, err := genkit.GenerateData[MenuItem](ctx, g,
 			ai.WithPrompt("Invent a menu item for a %s themed restaurant.", input.Theme),
 		)