add initializer

bgd-labs · Aug 14, 2024 · b651f7f · b651f7f
1 parent 0ecfd37
commit b651f7f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/periphery/contracts/static-a-token/StataMerger.sol b/src/periphery/contracts/static-a-token/StataMerger.sol
@@ -30,7 +30,7 @@ contract StataMerger is
     address newAToken,
     string calldata staticATokenName,
     string calldata staticATokenSymbol
-  ) external {
+  ) external initializer {
     __ERC20_init(staticATokenName, staticATokenSymbol);
     __ERC20Permit_init(staticATokenName);
     __ERC20AaveLM_init(newAToken);

diff --git a/tests/periphery/static-a-token/Stata4626LM.t.sol b/tests/periphery/static-a-token/Stata4626LM.t.sol
@@ -36,8 +36,8 @@ contract Stata4626LMTest is BaseTest {
     assertEq(staticATokenLM.name(), 'Static Aave Local WETH');
     assertEq(staticATokenLM.symbol(), 'stataLocWETH');
 
-    IERC20 aToken = staticATokenLM.aToken();
-    assertEq(address(aToken), A_TOKEN);
+    address referenceAsset = staticATokenLM.getReferenceAsset();
+    assertEq(referenceAsset, A_TOKEN);
 
     address underlyingAddress = address(staticATokenLM.asset());
     assertEq(underlyingAddress, UNDERLYING);