From 7074ab64270b49ea6128f5a5c41dfc3c10f8d4e6 Mon Sep 17 00:00:00 2001
From: Chepelau <chepelau.a@gmail.com>
Date: Fri, 25 Oct 2024 15:37:22 +0200
Subject: [PATCH] Fetch AMM owner address when testing ekubo pools

---
 tests/add_ekubo_pools.cairo | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/tests/add_ekubo_pools.cairo b/tests/add_ekubo_pools.cairo
index 6e36027..2d7192d 100644
--- a/tests/add_ekubo_pools.cairo
+++ b/tests/add_ekubo_pools.cairo
@@ -44,15 +44,12 @@ fn test_add_ekubo_options() {
         .unwrap();
 
     // Transfer ownership to governance
-    let amm_owner_address: ContractAddress =
-        0x74fd7da23e21f0f0479adb435221b23f57ca4c32a0c68aad9409a41c27f3067
-        .try_into()
-        .unwrap();
     let amm_address: ContractAddress =
         0x047472e6755afc57ada9550b6a3ac93129cc4b5f98f51c73e0644d129fd208d9
         .try_into()
         .unwrap();
     let amm = IAMMDispatcher { contract_address: amm_address };
+    let amm_owner_address: ContractAddress = amm.owner();
 
     prank(CheatTarget::One(amm_address), amm_owner_address, CheatSpan::TargetCalls(1));
     // start_prank(CheatTarget::One(amm_address), amm_owner_address);