From ecd933e8a5ffcb03fc86213508e3fcf4754be534 Mon Sep 17 00:00:00 2001
From: Alejandro <95312462+AGMASO@users.noreply.github.com>
Date: Mon, 5 Jan 2026 13:40:26 +0100
Subject: [PATCH] fix: fork enabled can send txs

---
 src/services/UmbrellaStakeDataService.ts | 32 ++++++++++++++----------
 1 file changed, 19 insertions(+), 13 deletions(-)

diff --git a/src/services/UmbrellaStakeDataService.ts b/src/services/UmbrellaStakeDataService.ts
index d26974f337..a3c4167e07 100644
--- a/src/services/UmbrellaStakeDataService.ts
+++ b/src/services/UmbrellaStakeDataService.ts
@@ -3,20 +3,26 @@ import { UmbrellaEthereum } from '@bgd-labs/aave-address-book';
 import { Provider } from '@ethersproject/providers';
 import { CustomMarket, MarketDataType } from 'src/ui-config/marketsConfig';
 
-type StakeUmbrellaConfig = Partial<{
-  [K in CustomMarket]: {
-    stakeDataProvider: string;
-    batchHelper: string;
-    stakeRewardsController: string;
-  };
-}>;
-
+type ForkedMarket = `fork_${CustomMarket}`;
+
+type StakeUmbrellaConfig = Partial<
+  Record<
+    CustomMarket | ForkedMarket,
+    {
+      stakeDataProvider: string;
+      batchHelper: string;
+      stakeRewardsController: string;
+    }
+  >
+>;
+const umbrellaMainnet = {
+  stakeDataProvider: '0x6321ba6b41fbddb6b678cd80db067f20a8770879',
+  batchHelper: UmbrellaEthereum.UMBRELLA_BATCH_HELPER,
+  stakeRewardsController: UmbrellaEthereum.UMBRELLA_REWARDS_CONTROLLER,
+};
 export const stakeUmbrellaConfig: StakeUmbrellaConfig = {
-  [CustomMarket.proto_mainnet_v3 || 'fork_proto_mainnet_v3']: {
-    stakeDataProvider: '0x6321ba6b41fbddb6b678cd80db067f20a8770879',
-    batchHelper: UmbrellaEthereum.UMBRELLA_BATCH_HELPER,
-    stakeRewardsController: UmbrellaEthereum.UMBRELLA_REWARDS_CONTROLLER,
-  },
+  [CustomMarket.proto_mainnet_v3]: umbrellaMainnet,
+  [`fork_${CustomMarket.proto_mainnet_v3}`]: umbrellaMainnet,
 };
 
 export class UmbrellaStakeDataService {