From 06a4cfe13b0f7938784188222b8e7a6c58f141be Mon Sep 17 00:00:00 2001
From: Sean Bowe <ewillbefull@gmail.com>
Date: Fri, 4 Sep 2020 04:49:59 -0600
Subject: [PATCH] Use extended omega for coset in prover.

---
 src/plonk/domain.rs | 4 ++++
 src/plonk/prover.rs | 2 +-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/plonk/domain.rs b/src/plonk/domain.rs
index 3203fb2..62ec88d 100644
--- a/src/plonk/domain.rs
+++ b/src/plonk/domain.rs
@@ -251,6 +251,10 @@ impl<G: Group> EvaluationDomain<G> {
         self.omega
     }
 
+    pub fn get_extended_omega(&self) -> G::Scalar {
+        self.extended_omega
+    }
+
     pub fn get_omega_inv(&self) -> G::Scalar {
         self.omega_inv
     }
diff --git a/src/plonk/prover.rs b/src/plonk/prover.rs
index d5a67e6..b4f2e07 100644
--- a/src/plonk/prover.rs
+++ b/src/plonk/prover.rs
@@ -330,7 +330,7 @@ impl<C: CurveAffine> Proof<C> {
 
             let mut right = permutation_product_cosets_inv[permutation_index].clone();
             let mut current_delta = x_0 * &C::Scalar::ZETA;
-            let step = domain.get_omega();
+            let step = domain.get_extended_omega();
             for advice in queries
                 .iter()
                 .map(|&query_index| &advice_cosets[query_index])