From 786826487cc7b1d350929c2e6960a52b16ef0227 Mon Sep 17 00:00:00 2001
From: Daniel Lubarov <daniel@lubarov.com>
Date: Mon, 19 Sep 2022 17:33:18 -0700
Subject: [PATCH] Update plonky2/src/fri/validate_shape.rs

Co-authored-by: wborgeaud <williamborgeaud@gmail.com>
---
 plonky2/src/fri/validate_shape.rs | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/plonky2/src/fri/validate_shape.rs b/plonky2/src/fri/validate_shape.rs
index e8a70a08..597adeda 100644
--- a/plonky2/src/fri/validate_shape.rs
+++ b/plonky2/src/fri/validate_shape.rs
@@ -40,7 +40,15 @@ where
             .iter()
             .zip(&instance.oracles)
         {
-            ensure!(leaf.len() == oracle.num_polys); // TODO: Account for blinding if ZK?
+            ensure!(
+                leaf.len()
+                    == oracle.num_polys
+                        + if oracle.blinding && params.hiding {
+                            SALT_SIZE
+                        } else {
+                            0
+                        }
+            );
             ensure!(merkle_proof.len() + cap_height == params.lde_bits());
         }