Don't unwrap on TxHasEnded

lambdaclass · Aug 13, 2024 · bea26df · bea26df
1 parent 4603b17
commit bea26df
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/core/lib/multivm/src/versions/era_vm/vm.rs b/core/lib/multivm/src/versions/era_vm/vm.rs
@@ -252,12 +252,12 @@ impl<S: ReadStorage + 'static> Vm<S> {
                         data.to_string()
                     };
 
-                    println!("BOOTLOADER: {} {}", msg, data_str)
+                    //println!("BOOTLOADER: {} {}", msg, data_str)
                 }
                 Hook::TxHasEnded => {
                     // println!("TX HAS ENDED");
-                    if let VmExecutionMode::OneTx = execution_mode {
-                        return self.last_tx_result.take().unwrap();
+                    if let (VmExecutionMode::OneTx, Some(result)) = (execution_mode, self.last_tx_result.take()) {
+                        return result;
                     }
                 }
             }