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])