diff --git a/lib/pipeline.ex b/lib/pipeline.ex
index c65bcd0..3538506 100644
--- a/lib/pipeline.ex
+++ b/lib/pipeline.ex
@@ -17,12 +17,12 @@ defmodule Loadfest.Pipeline do
     end
 
     def handle_demand(demand, state) when demand > 0 do
-      # num = Enum.random([1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 50, 50, 50, 250])
+      num = Enum.random([1, 50, 250])
 
       messages =
         for _i <- 1..demand do
           %Broadway.Message{
-            data: Loadfest.Worker.make_batch(1),
+            data: Loadfest.Worker.make_batch(num),
             acknowledger: {__MODULE__, :ack, 3}
           }
         end