diff --git a/contracts/mixins/FeesHelper.sol b/contracts/mixins/FeesHelper.sol index e97ae0982..7f7aa10ee 100644 --- a/contracts/mixins/FeesHelper.sol +++ b/contracts/mixins/FeesHelper.sol @@ -207,7 +207,7 @@ contract FeesHelper is State, FeesEvents { abi.encodeWithSelector( IPriceFeeds(_priceFeeds).queryReturn.selector, feeToken, - protocolTokenAddress, /// Price rewards using BZRX price rather than vesting token price. + sovTokenAddress, /// Price rewards using BZRX price rather than vesting token price. feeAmount.mul(_feeRebatePercent).div(10**20) ) ); @@ -218,16 +218,16 @@ contract FeesHelper is State, FeesEvents { } if (rewardAmount != 0) { - IERC20(protocolTokenAddress).approve(lockedSOVAddress, rewardAmount); + IERC20(sovTokenAddress).approve(lockedSOVAddress, rewardAmount); (bool success, ) = lockedSOVAddress.call(abi.encodeWithSignature("depositSOV(address,uint256)", user, rewardAmount)); if (success) { protocolTokenPaid = protocolTokenPaid.add(rewardAmount); - emit EarnReward(user, protocolTokenAddress, loanId, _feeRebatePercent, rewardAmount); + emit EarnReward(user, sovTokenAddress, loanId, _feeRebatePercent, rewardAmount); } else { - emit EarnRewardFail(user, protocolTokenAddress, loanId, _feeRebatePercent, rewardAmount); + emit EarnRewardFail(user, sovTokenAddress, loanId, _feeRebatePercent, rewardAmount); } } } diff --git a/tests-js/Utils/initializer.js b/tests-js/Utils/initializer.js index a69f751d8..64d790c7b 100644 --- a/tests-js/Utils/initializer.js +++ b/tests-js/Utils/initializer.js @@ -63,6 +63,7 @@ const getBZRX = async () => { const getSOV = async (sovryn, priceFeeds, SUSD, accounts) => { const sov = await TestToken.new("SOV", "SOV", 18, totalSupply); await sovryn.setProtocolTokenAddress(sov.address); + await sovryn.setSOVTokenAddress(sov.address); await sovryn.setLockedSOVAddress((await LockedSOVMockup.new(sov.address, [accounts[0]])).address); await priceFeeds.setRates(SUSD.address, sov.address, oneEth); diff --git a/tests-js/loan-token/LendingTestToken.test.js b/tests-js/loan-token/LendingTestToken.test.js index 5ece7d269..cb7d1298b 100644 --- a/tests-js/loan-token/LendingTestToken.test.js +++ b/tests-js/loan-token/LendingTestToken.test.js @@ -79,6 +79,7 @@ contract("LoanTokenLending", (accounts) => { await sovryn.setFeesController(lender); const sov = await TestToken.new("SOV", "SOV", 18, TOTAL_SUPPLY); await sovryn.setProtocolTokenAddress(sov.address); + await sovryn.setSOVTokenAddress(sov.address); await sovryn.setLockedSOVAddress((await LockedSOVMockup.new(sov.address, [accounts[0]])).address); loanTokenLogicStandard = await LoanTokenLogicStandard.new(); diff --git a/tests-js/loan-token/LendingwRBTCCollateral.test.js b/tests-js/loan-token/LendingwRBTCCollateral.test.js index 720f5bf01..e0b1e8c89 100644 --- a/tests-js/loan-token/LendingwRBTCCollateral.test.js +++ b/tests-js/loan-token/LendingwRBTCCollateral.test.js @@ -79,6 +79,7 @@ contract("LoanTokenLending", (accounts) => { tokenSOV = await SOV.new(TOTAL_SUPPLY); await sovryn.setLockedSOVAddress((await LockedSOVMockup.new(tokenSOV.address, [lender])).address); await sovryn.setProtocolTokenAddress(tokenSOV.address); + await sovryn.setSOVTokenAddress(tokenSOV.address); loanTokenLogicStandard = await LoanTokenLogicStandard.new(); loanToken = await LoanToken.new(lender, loanTokenLogicStandard.address, sovryn.address, rBTC.address); diff --git a/tests-js/loan-token/LendingwRBTCloan.test.js b/tests-js/loan-token/LendingwRBTCloan.test.js index b46fbe5f1..d9d5d7d73 100644 --- a/tests-js/loan-token/LendingwRBTCloan.test.js +++ b/tests-js/loan-token/LendingwRBTCloan.test.js @@ -89,6 +89,7 @@ contract("LoanTokenLending", (accounts) => { tokenSOV = await SOV.new(TOTAL_SUPPLY); await sovryn.setLockedSOVAddress((await LockedSOVMockup.new(tokenSOV.address, [lender])).address); await sovryn.setProtocolTokenAddress(tokenSOV.address); + await sovryn.setSOVTokenAddress(tokenSOV.address); loanTokenLogicWrbtc = await LoanTokenLogicWrbtc.new(); loanToken = await LoanToken.new(lender, loanTokenLogicWrbtc.address, sovryn.address, testWrbtc.address); diff --git a/tests-js/margin_trading_boilerplate.test.js b/tests-js/margin_trading_boilerplate.test.js index fd51f5049..8ff3e7ad7 100644 --- a/tests-js/margin_trading_boilerplate.test.js +++ b/tests-js/margin_trading_boilerplate.test.js @@ -57,6 +57,7 @@ contract("Margin Trading with Affiliates boilerplate", (accounts) => { const sov = await TestToken.new("SOV", "SOV", 18, web3.utils.toWei("20000", "ether")); await sovryn.setProtocolTokenAddress(sov.address); + await sovryn.setSOVTokenAddress(sov.address); await sovryn.setLockedSOVAddress((await LockedSOVMockup.new(sov.address, [accounts[0]])).address); loanToken = await LoanToken.new(owner, loanTokenLogic.address, sovryn.address, testWrbtc.address);