diff --git a/zero_bin/prover/src/lib.rs b/zero_bin/prover/src/lib.rs
index 37dcfaf19..def80ebcf 100644
--- a/zero_bin/prover/src/lib.rs
+++ b/zero_bin/prover/src/lib.rs
@@ -20,7 +20,9 @@ use zero_bin_common::fs::generate_block_proof_file_name;
 
 // Limit the number of segments that are proven in parallel
 // to prevent excessive memory consumption
+#[cfg(not(feature = "test_only"))]
 const GENERATION_INPUTS_AGG_CHUNK_SIZE_INNER: usize = 2;
+#[cfg(not(feature = "test_only"))]
 const GENERATION_INPUTS_AGG_CHUNK_SIZE: usize = 5;
 
 #[derive(Debug, Deserialize, Serialize)]