From e96fbdf29ce28a72c60cee0b08351997f2f505ac Mon Sep 17 00:00:00 2001
From: Marko Atanasievski <markoatana@gmail.com>
Date: Fri, 14 Jun 2024 13:54:05 +0200
Subject: [PATCH] fix: cleanup

---
 leader/src/jerigon.rs | 3 ++-
 leader/src/stdio.rs   | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/leader/src/jerigon.rs b/leader/src/jerigon.rs
index b845d0c0..d79a4b67 100644
--- a/leader/src/jerigon.rs
+++ b/leader/src/jerigon.rs
@@ -41,8 +41,9 @@ pub(crate) async fn jerigon_main(
             params.save_inputs_on_error,
             params.proof_output_dir.clone(),
         )
-        .await?;
+        .await;
     runtime.close().await?;
+    let proved_blocks = proved_blocks?;
 
     if params.keep_intermediate_proofs {
         warn!("Skipping cleanup, intermediate proofs are kept");
diff --git a/leader/src/stdio.rs b/leader/src/stdio.rs
index 22b44fa7..543979d3 100644
--- a/leader/src/stdio.rs
+++ b/leader/src/stdio.rs
@@ -21,8 +21,9 @@ pub(crate) async fn stdio_main(
 
     let proved_blocks = prover_input
         .prove(&runtime, previous, save_inputs_on_error, None)
-        .await?;
+        .await;
     runtime.close().await?;
+    let proved_blocks = proved_blocks?;
 
     let proofs: Vec<GeneratedBlockProof> = proved_blocks
         .into_iter()