From 164175d65efa5d4ee9112fa06fb69c50b1ba041b Mon Sep 17 00:00:00 2001
From: Robin Salen <salenrobin@gmail.com>
Date: Fri, 22 Sep 2023 12:34:56 -0400
Subject: [PATCH] Remove explicit panic

---
 evm/src/generation/outputs.rs | 2 +-
 evm/src/witness/errors.rs     | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/evm/src/generation/outputs.rs b/evm/src/generation/outputs.rs
index 2402d78cd4..60785e6dda 100644
--- a/evm/src/generation/outputs.rs
+++ b/evm/src/generation/outputs.rs
@@ -77,7 +77,7 @@ fn account_trie_record_to_output<F: Field>(
         .inputs
         .contract_code
         .get(&account.code_hash)
-        .unwrap_or_else(|| panic!("Code not found: {:?}", account.code_hash))
+        .ok_or_else(|| ProgramError::UnknownContractCode)?
         .clone();
 
     Ok(AccountOutput {
diff --git a/evm/src/witness/errors.rs b/evm/src/witness/errors.rs
index 2e543ac326..3ad10d17d7 100644
--- a/evm/src/witness/errors.rs
+++ b/evm/src/witness/errors.rs
@@ -16,6 +16,7 @@ pub enum ProgramError {
     InterpreterError,
     IntegerTooLarge,
     ProverInputError(ProverInputError),
+    UnknownContractCode,
 }
 
 #[allow(clippy::enum_variant_names)]