From b7e49d545827914af149a1995e40bb852301fea3 Mon Sep 17 00:00:00 2001
From: Thomas Kerber <thomas.kerber@iohk.io>
Date: Mon, 29 Apr 2024 12:42:24 +0100
Subject: [PATCH] Ensure downsize trims when required.

---
 halo2_proofs/src/poly/kzg/commitment.rs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/halo2_proofs/src/poly/kzg/commitment.rs b/halo2_proofs/src/poly/kzg/commitment.rs
index 011205c14b..7e0e919981 100644
--- a/halo2_proofs/src/poly/kzg/commitment.rs
+++ b/halo2_proofs/src/poly/kzg/commitment.rs
@@ -476,8 +476,12 @@ where
 
     fn downsize(&mut self, k: u32) {
         assert!(k <= self.k);
+        let n = 1 << k;
+        if n < self.trimed_size {
+            *self = self.clone().trim(n as usize);
+        }
         self.k = k;
-        self.n = 1 << k;
+        self.n = n as u64;
     }
 
     fn empty_msm(&'params self) -> MSMKZG<E> {