Fix L2 migration

safe-global · Dec 3, 2024 · 328191a · 328191a
1 parent 2a8697a
commit 328191a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/src/safe_cli/operators/safe_operator.py b/src/safe_cli/operators/safe_operator.py
@@ -726,9 +726,10 @@ def update_version_to_l2(
             fallback_handler = safe_deployments["1.3.0"][
                 "CompatibilityFallbackHandler"
             ][str(chain_id)]
+            # Assuming first element of the array is the `canonical` address
             data = HexBytes(
                 l2_migration_contract.functions.migrateFromV111(
-                    safe_l2_singleton, fallback_handler
+                    safe_l2_singleton[0], fallback_handler[0]
                 ).build_transaction(get_empty_tx_params())["data"]
             )
         elif safe_version in ("1.3.0", "1.4.1"):
@@ -738,7 +739,7 @@ def update_version_to_l2(
             fallback_handler = self.safe_cli_info.fallback_handler
             data = HexBytes(
                 l2_migration_contract.functions.migrateToL2(
-                    safe_l2_singleton
+                    safe_l2_singleton[0]
                 ).build_transaction(get_empty_tx_params())["data"]
             )
         else:

diff --git a/tests/test_safe_operator.py b/tests/test_safe_operator.py
@@ -371,9 +371,9 @@ def test_update_to_l2_v111(self):
             "safe_cli.operators.safe_operator.safe_deployments",
             {
                 "1.3.0": {
-                    "GnosisSafeL2": {"1337": self.safe_contract_V1_3_0.address},
+                    "GnosisSafeL2": {"1337": [self.safe_contract_V1_3_0.address]},
                     "CompatibilityFallbackHandler": {
-                        "1337": self.compatibility_fallback_handler.address
+                        "1337": [self.compatibility_fallback_handler.address]
                     },
                 }
             },
@@ -404,7 +404,7 @@ def test_update_to_l2_v130(self):
             "safe_cli.operators.safe_operator.safe_deployments",
             {
                 "1.3.0": {
-                    "GnosisSafeL2": {"1337": safe_contract_l2_130_address},
+                    "GnosisSafeL2": {"1337": [safe_contract_l2_130_address]},
                 }
             },
         ):