From 1ced853f33faea27f48b3484f8bfff895d0afceb Mon Sep 17 00:00:00 2001
From: wborgeaud <williamborgeaud@gmail.com>
Date: Tue, 12 Oct 2021 08:40:56 +0200
Subject: [PATCH] Add coset_index var

---
 src/plonk/get_challenges.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/plonk/get_challenges.rs b/src/plonk/get_challenges.rs
index 63802689..6a92c819 100644
--- a/src/plonk/get_challenges.rs
+++ b/src/plonk/get_challenges.rs
@@ -210,8 +210,8 @@ impl<F: RichField + Extendable<D>, const D: usize> CompressedProofWithPublicInpu
                 }
                 fri_inferred_elements.push(old_eval);
                 let arity = 1 << arity_bits;
-                let mut evals = self.proof.opening_proof.query_round_proofs.steps[i]
-                    [&(x_index >> arity_bits)]
+                let coset_index = x_index >> arity_bits;
+                let mut evals = self.proof.opening_proof.query_round_proofs.steps[i][&coset_index]
                     .evals
                     .clone();
                 let x_index_within_coset = x_index & (arity - 1);