From 8777c135fb5dff54a8288a6403c0d014e83b4c09 Mon Sep 17 00:00:00 2001
From: evalir <hi@enriqueortiz.dev>
Date: Wed, 3 Jan 2024 11:22:04 -0400
Subject: [PATCH] fix(cheatcodes): always omit pk on addr (#6698)

---
 crates/evm/traces/src/decoder/mod.rs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/crates/evm/traces/src/decoder/mod.rs b/crates/evm/traces/src/decoder/mod.rs
index c13ce494a1a8e..e37880249e74e 100644
--- a/crates/evm/traces/src/decoder/mod.rs
+++ b/crates/evm/traces/src/decoder/mod.rs
@@ -357,7 +357,7 @@ impl CallTraceDecoder {
     fn decode_cheatcode_inputs(&self, func: &Function, data: &[u8]) -> Option<Vec<String>> {
         match func.name.as_str() {
             "expectRevert" => Some(vec![decode::decode_revert(data, Some(&self.errors), None)]),
-            "rememberKey" | "addr" | "startBroadcast" | "broadcast" => {
+            "rememberKey" | "startBroadcast" | "broadcast" => {
                 // these functions accept a private key as uint256, which should not be
                 // converted to plain text
                 if !func.inputs.is_empty() && func.inputs[0].ty != "uint256" {
@@ -375,6 +375,7 @@ impl CallTraceDecoder {
                 }
                 Some(decoded.iter().map(format_token).collect())
             }
+            "addr" => Some(vec!["<pk>".to_string()]),
             "deriveKey" => Some(vec!["<pk>".to_string()]),
             "parseJson" |
             "parseJsonUint" |