diff --git a/examples/text-generation/run_generation.py b/examples/text-generation/run_generation.py
index cdfdd614e9..abce2bd963 100755
--- a/examples/text-generation/run_generation.py
+++ b/examples/text-generation/run_generation.py
@@ -569,7 +569,7 @@ def rounder(x):
             for j, output in enumerate(
                 zip(generated[args.num_return_sequences * i : args.num_return_sequences * (i + 1)])
             ):
-                print(f"output {j+1}: {output}")
+                print(f"output {i+1}.{j+1}: {output}")
                 all_outputs.append(output)
             print()