diff --git a/examples/handoffs/message_filter.py b/examples/handoffs/message_filter.py
index 9dd56ef7..b7fed6c1 100644
--- a/examples/handoffs/message_filter.py
+++ b/examples/handoffs/message_filter.py
@@ -60,9 +60,9 @@ async def main():
 
         print("Step 1 done")
 
-        # 2. Ask it to square a number
+        # 2. Ask it to generate a number
         result = await Runner.run(
-            second_agent,
+            first_agent,
             input=result.to_input_list()
             + [{"content": "Can you generate a random number between 0 and 100?", "role": "user"}],
         )
diff --git a/examples/handoffs/message_filter_streaming.py b/examples/handoffs/message_filter_streaming.py
index 8d1b4208..63cb1de3 100644
--- a/examples/handoffs/message_filter_streaming.py
+++ b/examples/handoffs/message_filter_streaming.py
@@ -60,9 +60,9 @@ async def main():
 
         print("Step 1 done")
 
-        # 2. Ask it to square a number
+        # 2. Ask it to generate a number
         result = await Runner.run(
-            second_agent,
+            first_agent,
             input=result.to_input_list()
             + [{"content": "Can you generate a random number between 0 and 100?", "role": "user"}],
         )