diff --git a/src/prover.rs b/src/prover.rs
index 434d0084..cb2e963e 100644
--- a/src/prover.rs
+++ b/src/prover.rs
@@ -118,9 +118,7 @@ pub(crate) fn prove<F: Field>(
 
     challenger.observe_hash(&quotient_polys_commitment.merkle_tree.root);
 
-    // TODO: How many do we need?
-    let num_zetas = 2;
-    let zetas = challenger.get_n_challenges(num_zetas);
+    let zetas = challenger.get_n_challenges(config.num_checks);
 
     let (opening_proof, openings) = timed!(
         ListPolynomialCommitment::batch_open_plonk(