Add payable to aaveSupply function

EridianAlpha · Jun 17, 2024 · 0ea8097 · 0ea8097
1 parent 51fe0c0
commit 0ea8097
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/AavePM.sol b/src/AavePM.sol
@@ -369,7 +369,12 @@ contract AavePM is
     }
 
     /// @notice // TODO: Add comment
-    function aaveSupplyFromContractBalance() public onlyRole(MANAGER_ROLE) returns (uint256 suppliedCollateral) {
+    function aaveSupplyFromContractBalance()
+        public
+        payable
+        onlyRole(MANAGER_ROLE)
+        returns (uint256 suppliedCollateral)
+    {
         suppliedCollateral = abi.decode(
             delegateCallHelper(
                 "aaveFunctionsModule",

diff --git a/src/interfaces/IAavePM.sol b/src/interfaces/IAavePM.sol
@@ -102,7 +102,7 @@ interface IAavePM {
     function rebalance() external returns (uint256 repaymentAmountUSDC);
     function reinvest() external returns (uint256 reinvestedDebt);
     function deleverage() external;
-    function aaveSupplyFromContractBalance() external returns (uint256 suppliedCollateral);
+    function aaveSupplyFromContractBalance() external payable returns (uint256 suppliedCollateral);
     function aaveRepayUSDCFromContractBalance() external;
 
     /// TODO: Move to different heading