From 66a8134a515c9f699d572e7f681311a04df3aef8 Mon Sep 17 00:00:00 2001
From: Xinding Wei <weixinding@gmail.com>
Date: Thu, 16 Jan 2025 01:08:09 +0800
Subject: [PATCH] Fix Stateful for FriReducedOpeningChip (#1218)

---
 extensions/native/circuit/src/fri/mod.rs | 1 +
 1 file changed, 1 insertion(+)

diff --git a/extensions/native/circuit/src/fri/mod.rs b/extensions/native/circuit/src/fri/mod.rs
index eedb919b86..d50ab99934 100644
--- a/extensions/native/circuit/src/fri/mod.rs
+++ b/extensions/native/circuit/src/fri/mod.rs
@@ -593,6 +593,7 @@ where
 impl<F: PrimeField32> Stateful<Vec<u8>> for FriReducedOpeningChip<F> {
     fn load_state(&mut self, state: Vec<u8>) {
         self.records = bitcode::deserialize(&state).unwrap();
+        self.height = self.records.iter().map(|record| record.a_reads.len()).sum();
     }
 
     fn store_state(&self) -> Vec<u8> {