From 1808db7788ff19bb0212c3e55d817dd9c2e4e048 Mon Sep 17 00:00:00 2001
From: Christophe <contact@christophedeveaux.com>
Date: Wed, 10 Jul 2024 15:51:27 +0000
Subject: [PATCH] Update cache key

---
 src/getRollupAddress.ts | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/getRollupAddress.ts b/src/getRollupAddress.ts
index 532d0e8a..8adf7a7e 100644
--- a/src/getRollupAddress.ts
+++ b/src/getRollupAddress.ts
@@ -1,7 +1,7 @@
 import { Address, Chain, PublicClient, Transport } from 'viem';
 import { sequencerInboxABI } from './abi';
 
-const cache: Record<Address, Address> = {};
+const cache: Record<string, Address> = {};
 export async function getRollupAddress<TChain extends Chain | undefined>(
   publicClient: PublicClient<Transport, TChain>,
   params: { sequencerInbox: Address } | { rollupAdminLogic: Address },
@@ -11,7 +11,7 @@ export async function getRollupAddress<TChain extends Chain | undefined>(
     return params.rollupAdminLogic;
   }
 
-  const addressFromCache = cache[params.sequencerInbox];
+  const addressFromCache = cache[`${publicClient.chain?.id}_${params.sequencerInbox}`];
   if (addressFromCache) {
     return addressFromCache;
   }
@@ -22,6 +22,6 @@ export async function getRollupAddress<TChain extends Chain | undefined>(
     address: params.sequencerInbox,
     abi: sequencerInboxABI,
   });
-  cache[params.sequencerInbox] = rollupAddress;
+  cache[`${publicClient.chain?.id}_${params.sequencerInbox}`] = rollupAddress;
   return rollupAddress;
 }