diff --git a/src/cli/playbook/ipa.rs b/src/cli/playbook/ipa.rs
index 1f672d75c..3c758a6a9 100644
--- a/src/cli/playbook/ipa.rs
+++ b/src/cli/playbook/ipa.rs
@@ -100,7 +100,7 @@ where
     let inputs = buffers.map(BodyStream::from);
     tracing::info!("Starting query after finishing encryption");
 
-    do_processing::<F>(inputs, clients, query_id, query_config).await
+    do_processing::<F>(inputs, query_size, clients, query_id, query_config).await
 }
 
 pub async fn playbook_oprf_ipa<F>(
@@ -136,11 +136,12 @@ where
     let inputs = buffers.map(BodyStream::from);
     tracing::info!("Starting query for OPRF");
 
-    do_processing::<F>(inputs, clients, query_id, query_config).await
+    do_processing::<F>(inputs, query_size, clients, query_id, query_config).await
 }
 
 pub async fn do_processing<F>(
     inputs: [BodyStream; 3],
+    query_size: usize,
     clients: &[MpcHelperClient; 3],
     query_id: QueryId,
     query_config: IpaQueryConfig,
@@ -149,7 +150,6 @@ where
     F: PrimeField,
     AdditiveShare<F>: Serializable,
 {
-    let query_size = inputs.len();
     let mpc_time = Instant::now();
     try_join_all(
         inputs