diff --git a/crates/cestory/src/lib.rs b/crates/cestory/src/lib.rs index 60c1a4e3..f9e0f07d 100644 --- a/crates/cestory/src/lib.rs +++ b/crates/cestory/src/lib.rs @@ -14,7 +14,7 @@ use serde::{ Deserialize, Deserializer, Serialize, Serializer, }; -use crate::{light_validation::LightValidation, system::System}; +use crate::light_validation::LightValidation; use anyhow::{anyhow, Context as _, Result}; use ces_crypto::{ aead, @@ -314,6 +314,7 @@ impl Ceseal { if let Some(system) = &mut self.system { system.sealing_path = self.args.sealing_path.clone(); system.storage_path = self.args.storage_path.clone(); + system.args = self.args.clone(); } } @@ -599,14 +600,11 @@ impl Ceseal { let recv_mq = &mut runtime_state.recv_mq; let send_mq = &mut runtime_state.send_mq; let seq = &mut seq; - let mut system: System = - ces_mq::checkpoint_helper::using_dispatcher(recv_mq, move || { - ces_mq::checkpoint_helper::using_send_mq(send_mq, || { - seq.next_element()?.ok_or_else(|| de::Error::custom("Missing System")) - }) - })?; - system.args = factory.args.clone(); - Some(system) + ces_mq::checkpoint_helper::using_dispatcher(recv_mq, move || { + ces_mq::checkpoint_helper::using_send_mq(send_mq, || { + seq.next_element()?.ok_or_else(|| de::Error::custom("Missing System")) + }) + })? }; } else { let _: Option = seq.next_element()?;