Hardcode DAI address in the fee collection functions

OasisDEX · Nov 15, 2021 · f3ef0fb · f3ef0fb
1 parent 158788c
commit f3ef0fb
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/contracts/multiply/Exchange.sol b/contracts/multiply/Exchange.sol
@@ -92,16 +92,16 @@ contract Exchange {
     return balance;
   }
 
-  function _collectFeeTokenToDai(address asset, uint256 fromAmount) internal returns (uint256) {
+  function _collectFeeTokenToDai(uint256 fromAmount) internal returns (uint256) {
     uint256 feeToTransfer = (fromAmount.mul(fee)).div(feeBase);
-    IERC20(asset).safeTransfer(feeBeneficiaryAddress, feeToTransfer);
+    IERC20(DAI_ADDRESS).safeTransfer(feeBeneficiaryAddress, feeToTransfer);
     emit FeePaid(feeBeneficiaryAddress, feeToTransfer);
     return fromAmount.sub(feeToTransfer);
   }
 
-  function _collectFeeDaiToToken(address asset, uint256 fromAmount) internal returns (uint256) {
+  function _collectFeeDaiToToken(uint256 fromAmount) internal returns (uint256) {
     uint256 feeToTransfer = fromAmount.sub((fromAmount.mul(feeBase)).div(feeBase.add(fee)));
-    IERC20(asset).safeTransfer(feeBeneficiaryAddress, feeToTransfer);
+    IERC20(DAI_ADDRESS).safeTransfer(feeBeneficiaryAddress, feeToTransfer);
     emit FeePaid(feeBeneficiaryAddress, feeToTransfer);
     return fromAmount.sub(feeToTransfer);
   }
@@ -123,7 +123,7 @@ contract Exchange {
   ) public {
     _transferIn(msg.sender, DAI_ADDRESS, amount);
 
-    uint256 _amount = _collectFeeDaiToToken(DAI_ADDRESS, amount);
+    uint256 _amount = _collectFeeDaiToToken(amount);
     uint256 balance = _swap(DAI_ADDRESS, asset, _amount, receiveAtLeast, callee, withData);
 
     uint256 daiBalance = IERC20(DAI_ADDRESS).balanceOf(address(this));
@@ -145,7 +145,7 @@ contract Exchange {
     _transferIn(msg.sender, asset, amount);
 
     uint256 balance = _swap(asset, DAI_ADDRESS, amount, receiveAtLeast, callee, withData);
-    uint256 _balance = _collectFeeTokenToDai(DAI_ADDRESS, balance);
+    uint256 _balance = _collectFeeTokenToDai(balance);
 
     uint256 assetBalance = IERC20(asset).balanceOf(address(this));