From e690e3d82b0f1a8803942236253165fef2779be3 Mon Sep 17 00:00:00 2001
From: Grezle <4310551+grezle@users.noreply.github.com>
Date: Thu, 25 Sep 2025 13:49:20 +0100
Subject: [PATCH 1/2] feat: use appropriate zeusEnv for contextName

---
 pkg/common/zeus.go | 18 ++++++++++++++----
 1 file changed, 14 insertions(+), 4 deletions(-)

diff --git a/pkg/common/zeus.go b/pkg/common/zeus.go
index 43d71516..257e8ef9 100644
--- a/pkg/common/zeus.go
+++ b/pkg/common/zeus.go
@@ -35,20 +35,30 @@ type L2ZeusAddressData struct {
 }
 
 // GetZeusAddresses runs the zeus env show commands and extracts core EigenLayer addresses.
-func GetZeusAddresses(ctx context.Context, logger iface.Logger) (*L1ZeusAddressData, *L2ZeusAddressData, error) {
+func GetZeusAddresses(ctx context.Context, logger iface.Logger, contextName string) (*L1ZeusAddressData, *L2ZeusAddressData, error) {
 	var (
 		l1Raw, l2Raw []byte
 		err          error
 	)
 
+	// Default zeus env to sepolia for testnet/devnet
+	l1ZeusEnv := "testnet-sepolia"
+	l2ZeusEnv := "testnet-base-sepolia"
+
+	// Overide with mainnet envs
+	if contextName == "mainnet" {
+		l1ZeusEnv = "mainnet"
+		l2ZeusEnv = "base"
+	}
+
 	// Run L1
-	l1Raw, err = runZeusJSON(ctx, "testnet-sepolia")
+	l1Raw, err = runZeusJSON(ctx, l1ZeusEnv)
 	if err != nil {
 		return nil, nil, fmt.Errorf("zeus L1: %w", err)
 	}
 
 	// Run L2
-	l2Raw, err = runZeusJSON(ctx, "testnet-base-sepolia")
+	l2Raw, err = runZeusJSON(ctx, l2ZeusEnv)
 	if err != nil {
 		return nil, nil, fmt.Errorf("zeus L2: %w", err)
 	}
@@ -126,7 +136,7 @@ func GetZeusAddresses(ctx context.Context, logger iface.Logger) (*L1ZeusAddressD
 // UpdateContextWithZeusAddresses updates the context configuration with addresses from Zeus
 func UpdateContextWithZeusAddresses(context context.Context, logger iface.Logger, contextMap *yaml.Node, contextName string) error {
 	logger.Title("Fetching EigenLayer core addresses for L1 and L2 from Zeus...")
-	l1Addresses, l2Addresses, err := GetZeusAddresses(context, logger)
+	l1Addresses, l2Addresses, err := GetZeusAddresses(context, logger, contextName)
 	if err != nil {
 		return err
 	}

From 5291ee06324be7026e9cf6fc2e0326eec5a214d1 Mon Sep 17 00:00:00 2001
From: Grezle <4310551+grezle@users.noreply.github.com>
Date: Thu, 25 Sep 2025 16:16:01 +0100
Subject: [PATCH 2/2] fix: typo

---
 pkg/common/zeus.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pkg/common/zeus.go b/pkg/common/zeus.go
index 257e8ef9..f981b731 100644
--- a/pkg/common/zeus.go
+++ b/pkg/common/zeus.go
@@ -45,7 +45,7 @@ func GetZeusAddresses(ctx context.Context, logger iface.Logger, contextName stri
 	l1ZeusEnv := "testnet-sepolia"
 	l2ZeusEnv := "testnet-base-sepolia"
 
-	// Overide with mainnet envs
+	// Override with mainnet envs
 	if contextName == "mainnet" {
 		l1ZeusEnv = "mainnet"
 		l2ZeusEnv = "base"