From f61d7c0f715a550223f6faf75f70cae3cd8d670b Mon Sep 17 00:00:00 2001
From: Sai Deng <deng.sai@gmail.com>
Date: Wed, 2 Oct 2024 14:55:37 -0700
Subject: [PATCH] fix

---
 evm_arithmetization/src/verifier.rs | 21 ++++-----------------
 1 file changed, 4 insertions(+), 17 deletions(-)

diff --git a/evm_arithmetization/src/verifier.rs b/evm_arithmetization/src/verifier.rs
index b53c2b077..9d1ac5f03 100644
--- a/evm_arithmetization/src/verifier.rs
+++ b/evm_arithmetization/src/verifier.rs
@@ -120,14 +120,13 @@ fn verify_proof<F: RichField + Extendable<D>, C: GenericConfig<D, F = F>, const
     all_stark: &AllStark<F, D>,
     all_proof: AllProof<F, C, D>,
     config: &StarkConfig,
-    use_keccak_tables: bool,
     is_initial: bool,
 ) -> Result<()> {
     let AllProofChallenges {
         stark_challenges,
         ctl_challenges,
     } = all_proof
-        .get_challenges(config, use_keccak_tables)
+        .get_challenges(config, all_proof.use_keccak_tables)
         .map_err(|_| anyhow::Error::msg("Invalid sampling of proof challenges."))?;
 
     let num_lookup_columns = all_stark.num_lookups_helper_columns(config);
@@ -175,7 +174,7 @@ fn verify_proof<F: RichField + Extendable<D>, C: GenericConfig<D, F = F>, const
     verify_table!(arithmetic_stark, Table::Arithmetic);
     verify_table!(byte_packing_stark, Table::BytePacking);
     verify_table!(cpu_stark, Table::Cpu);
-    if use_keccak_tables {
+    if all_proof.use_keccak_tables {
         verify_table!(keccak_stark, Table::Keccak);
         verify_table!(keccak_sponge_stark, Table::KeccakSponge);
     }
@@ -430,22 +429,10 @@ pub mod testing {
 
         // TODO(sdeng): determine the value of `use_keccak_tables` from `all_proof`.
         let use_keccak_tables = true;
-        verify_proof(
-            all_stark,
-            all_proofs[0].clone(),
-            config,
-            use_keccak_tables,
-            true,
-        )?;
+        verify_proof(all_stark, all_proofs[0].clone(), config, use_keccak_tables)?;
 
         for all_proof in &all_proofs[1..] {
-            verify_proof(
-                all_stark,
-                all_proof.clone(),
-                config,
-                use_keccak_tables,
-                false,
-            )?;
+            verify_proof(all_stark, all_proof.clone(), config, use_keccak_tables)?;
         }
 
         Ok(())