fix: cleanup

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");

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()