diff --git a/evm_arithmetization/src/fixed_recursive_verifier.rs b/evm_arithmetization/src/fixed_recursive_verifier.rs
index 37415aa39..7cc105d44 100644
--- a/evm_arithmetization/src/fixed_recursive_verifier.rs
+++ b/evm_arithmetization/src/fixed_recursive_verifier.rs
@@ -3181,7 +3181,7 @@ mod tests {
             all_circuits.prove_segment(
                 &all_stark,
                 &config,
-                payload.trim(),
+                payload,
                 &mut segment_data,
                 timing,
                 None,
diff --git a/evm_arithmetization/src/testing_utils.rs b/evm_arithmetization/src/testing_utils.rs
index 26b98a3b8..dc6460f7e 100644
--- a/evm_arithmetization/src/testing_utils.rs
+++ b/evm_arithmetization/src/testing_utils.rs
@@ -14,7 +14,7 @@ use plonky2::field::goldilocks_field::GoldilocksField;
 
 pub use crate::cpu::kernel::cancun_constants::*;
 pub use crate::cpu::kernel::constants::global_exit_root::*;
-use crate::generation::TrieInputs;
+use crate::generation::{TrieInputs, TrimmedGenerationInputs};
 use crate::proof::TrieRoots;
 use crate::witness::operation::Operation;
 use crate::{
@@ -219,15 +219,18 @@ fn empty_payload() -> Result<GenerationInputs> {
     Ok(inputs)
 }
 
-pub fn segment_without_keccak() -> Result<(GenerationInputs, GenerationSegmentData)> {
+pub fn segment_without_keccak() -> Result<(
+    TrimmedGenerationInputs<GoldilocksField>,
+    GenerationSegmentData,
+)> {
     let payload = empty_payload()?;
     let max_cpu_len_log = Some(7);
     let mut segment_iterator =
         SegmentDataIterator::<GoldilocksField>::new(&payload, max_cpu_len_log);
-    let (_, segment_data) = segment_iterator.next().unwrap()?;
+    let (trimmed_inputs, segment_data) = segment_iterator.next().unwrap()?;
 
     let opcode_counts = &segment_data.opcode_counts;
     assert!(!opcode_counts.contains_key(&Operation::KeccakGeneral));
 
-    Ok((payload, segment_data))
+    Ok((trimmed_inputs, segment_data))
 }
diff --git a/evm_arithmetization/tests/add11_yml.rs b/evm_arithmetization/tests/add11_yml.rs
index e39762a49..dd9bfb1ce 100644
--- a/evm_arithmetization/tests/add11_yml.rs
+++ b/evm_arithmetization/tests/add11_yml.rs
@@ -110,7 +110,7 @@ fn get_generation_inputs() -> GenerationInputs {
             block_metadata.block_timestamp,
             block_metadata.parent_beacon_block_root,
         )
-            .unwrap();
+        .unwrap();
         let beacon_roots_account =
             beacon_roots_contract_from_storage(&beacon_roots_account_storage);
 
@@ -131,7 +131,7 @@ fn get_generation_inputs() -> GenerationInputs {
                 nibbles: Nibbles::from_h256_be(keccak([0u8; 32])),
                 value: vec![2],
             })
-                .hash(),
+            .hash(),
             ..AccountRlp::default()
         };
 
@@ -174,7 +174,7 @@ fn get_generation_inputs() -> GenerationInputs {
         nibbles: Nibbles::from_str("0x80").unwrap(),
         value: txn.to_vec(),
     }
-        .into();
+    .into();
 
     let trie_roots_after = TrieRoots {
         state_root: expected_state_trie_after.hash(),
diff --git a/evm_arithmetization/tests/empty_tables.rs b/evm_arithmetization/tests/empty_tables.rs
index 3d24ccee0..9cb3e6fa9 100644
--- a/evm_arithmetization/tests/empty_tables.rs
+++ b/evm_arithmetization/tests/empty_tables.rs
@@ -29,7 +29,7 @@ fn empty_tables() -> anyhow::Result<()> {
     let proof = prove::<F, C, D>(
         &all_stark,
         &config,
-        payload.trim(),
+        payload,
         &mut segment_data,
         &mut timing,
         None,