diff --git a/samples/controlled_generation.py b/samples/controlled_generation.py
index e46b1a912..b0c269bb7 100644
--- a/samples/controlled_generation.py
+++ b/samples/controlled_generation.py
@@ -27,7 +27,7 @@ class Recipe(typing.TypedDict):
         result = model.generate_content(
             "List a few popular cookie recipes.",
             generation_config=genai.GenerationConfig(
-                response_mime_type="application/json", response_schema=list([Recipe])
+                response_mime_type="application/json", response_schema=list[Recipe]
             ),
         )
         print(result)