@ensuro/core 2.0.0-beta8 → 2.0.0

package/contracts/PolicyPoolComponent.sol

@@ -3,19 +3,20 @@ pragma solidity ^0.8.0;
 
 import {UUPSUpgradeable} from "@openzeppelin/contracts-upgradeable/proxy/utils/UUPSUpgradeable.sol";
 import {IERC20Metadata} from "@openzeppelin/contracts/token/ERC20/extensions/IERC20Metadata.sol";
+import {IERC165} from "@openzeppelin/contracts/utils/introspection/IERC165.sol";
 import {IPolicyPool} from "./interfaces/IPolicyPool.sol";
 import {IPolicyPoolComponent} from "./interfaces/IPolicyPoolComponent.sol";
 import {IAccessManager} from "./interfaces/IAccessManager.sol";
 import {PausableUpgradeable} from "@openzeppelin/contracts-upgradeable/security/PausableUpgradeable.sol";
-import {WadRayMath} from "./WadRayMath.sol";
+import {WadRayMath} from "./dependencies/WadRayMath.sol";
 
 /**
  * @title Base class for PolicyPool components
  * @dev This is the base class of all the components of the protocol that are linked to the PolicyPool and created
  *      after it.
  *      Holds the reference to _policyPool as immutable, also provides access to common admin roles:
- *      - LEVEL1_ROLE: High impact changes like upgrades or other critical operations
- *      - LEVEL2_ROLE: Mid-impact changes like adding new risk modules or changing some parameters
+ *      - LEVEL1_ROLE: High impact changes like upgrades, adding or removing components or other critical operations
+ *      - LEVEL2_ROLE: Mid-impact changes like changing some parameters
  *      - LEVEL3_ROLE: Low-impact changes like changing some parameters up to given percentage (tweaks)
  *      - GUARDIAN_ROLE: For emergency operations oriented to protect the protocol in case of attacks or hacking.
  *
@@ -51,17 +52,17 @@ abstract contract PolicyPoolComponent is
   }
 
   modifier onlyComponentRole(bytes32 role) {
-    _policyPool.access().checkComponentRole(address(this), role, msg.sender, false);
+    _policyPool.access().checkComponentRole(address(this), role, _msgSender(), false);
     _;
   }
 
   modifier onlyGlobalOrComponentRole(bytes32 role) {
-    _policyPool.access().checkComponentRole(address(this), role, msg.sender, true);
+    _policyPool.access().checkComponentRole(address(this), role, _msgSender(), true);
     _;
   }
 
   modifier onlyGlobalOrComponentRole2(bytes32 role1, bytes32 role2) {
-    _policyPool.access().checkComponentRole2(address(this), role1, role2, msg.sender, true);
+    _policyPool.access().checkComponentRole2(address(this), role1, role2, _msgSender(), true);
     _;
   }
 
@@ -71,19 +72,24 @@ abstract contract PolicyPoolComponent is
     bytes32 role3
   ) {
     IAccessManager access = _policyPool.access();
-    if (!access.hasComponentRole(address(this), role1, msg.sender, true)) {
-      _policyPool.access().checkComponentRole2(address(this), role2, role3, msg.sender, true);
+    if (!access.hasComponentRole(address(this), role1, _msgSender(), true)) {
+      _policyPool.access().checkComponentRole2(address(this), role2, role3, _msgSender(), true);
     }
     _;
   }
 
   /// @custom:oz-upgrades-unsafe-allow constructor
   constructor(IPolicyPool policyPool_) {
+    require(
+      address(policyPool_) != address(0),
+      "PolicyPoolComponent: policyPool cannot be zero address"
+    );
+    _disableInitializers();
     _policyPool = policyPool_;
   }
 
   // solhint-disable-next-line func-name-mixedcase
-  function __PolicyPoolComponent_init() internal initializer {
+  function __PolicyPoolComponent_init() internal onlyInitializing {
     __UUPSUpgradeable_init();
     __Pausable_init();
   }
@@ -94,12 +100,25 @@ abstract contract PolicyPoolComponent is
     override
     onlyGlobalOrComponentRole2(GUARDIAN_ROLE, LEVEL1_ROLE)
   {
+    _upgradeValidations(newImpl);
+  }
+
+  function _upgradeValidations(address newImpl) internal view virtual {
     require(
       IPolicyPoolComponent(newImpl).policyPool() == _policyPool,
       "Can't upgrade changing the PolicyPool!"
     );
   }
 
+  /**
+   * @dev See {IERC165-supportsInterface}.
+   */
+  function supportsInterface(bytes4 interfaceId) public view virtual override returns (bool) {
+    return
+      interfaceId == type(IERC165).interfaceId ||
+      interfaceId == type(IPolicyPoolComponent).interfaceId;
+  }
+
   function pause() public onlyGlobalOrComponentRole(GUARDIAN_ROLE) {
     _pause();
   }
@@ -117,22 +136,7 @@ abstract contract PolicyPoolComponent is
   }
 
   function hasPoolRole(bytes32 role) internal view returns (bool) {
-    return _policyPool.access().hasComponentRole(address(this), role, msg.sender, true);
-  }
-
-  function _isTweakRay(
-    uint256 oldValue,
-    uint256 newValue,
-    uint256 maxTweak
-  ) internal pure returns (bool) {
-    if (oldValue == newValue) return true;
-    if (oldValue == 0) return maxTweak >= WadRayMath.RAY;
-    if (newValue == 0) return false;
-    if (oldValue < newValue) {
-      return (newValue.rayDiv(oldValue) - WadRayMath.RAY) <= maxTweak;
-    } else {
-      return (WadRayMath.RAY - newValue.rayDiv(oldValue)) <= maxTweak;
-    }
+    return _policyPool.access().hasComponentRole(address(this), role, _msgSender(), true);
   }
 
   function _isTweakWad(
@@ -186,4 +190,11 @@ abstract contract PolicyPoolComponent is
       }
     }
   }
+
+  /**
+   * @dev This empty reserved space is put in place to allow future versions to add new
+   * variables without shifting down storage in the inheritance chain.
+   * See https://docs.openzeppelin.com/contracts/4.x/upgradeable#storage_gaps
+   */
+  uint256[49] private __gap;
 }

package/contracts/PremiumsAccount.sol

@@ -4,10 +4,12 @@ pragma solidity ^0.8.0;
 import {Math} from "@openzeppelin/contracts/utils/math/Math.sol";
 import {IERC20Metadata} from "@openzeppelin/contracts/token/ERC20/extensions/IERC20Metadata.sol";
 import {SafeERC20} from "@openzeppelin/contracts/token/ERC20/utils/SafeERC20.sol";
-import {WadRayMath} from "./WadRayMath.sol";
+import {SafeCast} from "@openzeppelin/contracts/utils/math/SafeCast.sol";
+import {WadRayMath} from "./dependencies/WadRayMath.sol";
 import {IPolicyPool} from "./interfaces/IPolicyPool.sol";
 import {IEToken} from "./interfaces/IEToken.sol";
 import {Reserve} from "./Reserve.sol";
+import {IAccessManager} from "./interfaces/IAccessManager.sol";
 import {IPremiumsAccount} from "./interfaces/IPremiumsAccount.sol";
 import {Policy} from "./Policy.sol";
 import {IEToken} from "./interfaces/IEToken.sol";
@@ -15,7 +17,14 @@ import {IAssetManager} from "./interfaces/IAssetManager.sol";
 
 /**
  * @title Ensuro Premiums Account
- * @dev This contract holds the premiums of a set of risk modules
+ * @dev This contract holds the pure premiums of a set of risk modules. The pure premiums is the part of the premium
+ * that is expected to cover the losses. The contract keeps track of the pure premiums of the active policies
+ * (_activePurePremiums) and the surplus or deficit generated by the finalized policies (pure premiums collected -
+ * losses).
+ *
+ * Collaborates with a junior {EToken} and a senior {EToken} that act as lenders when the premiums aren't enough to
+ * cover the losses.
+ *
  * @custom:security-contact security@ensuro.co
  * @author Ensuro
  */
@@ -23,15 +32,37 @@ contract PremiumsAccount is IPremiumsAccount, Reserve {
   using Policy for Policy.PolicyData;
   using WadRayMath for uint256;
   using SafeERC20 for IERC20Metadata;
+  using SafeCast for uint256;
 
   bytes32 public constant WITHDRAW_WON_PREMIUMS_ROLE = keccak256("WITHDRAW_WON_PREMIUMS_ROLE");
+  uint256 internal constant FOUR_DECIMAL_TO_WAD = 1e14;
+  uint16 internal constant HUNDRED_PERCENT = 1e4;
 
+  /**
+   * @dev The Junior eToken is the first {EToken} to which the PremiumsAccount will go for credit when it runs out of
+   * money. Optional (address(0)).
+   */
   /// @custom:oz-upgrades-unsafe-allow state-variable-immutable
   IEToken internal immutable _juniorEtk;
+
+  /**
+   * @dev The Senior eToken is the second {EToken} to which the PremiumsAccount will go for credit, after trying before
+   * with the junior eToken, when it runs out of money. Optional (address(0)).
+   */
   /// @custom:oz-upgrades-unsafe-allow state-variable-immutable
   IEToken internal immutable _seniorEtk;
 
+  /**
+   * @dev The active pure premiums field keeps track of the pure premiums collected by the active policies of risk
+   * modules linked with this PremiumsAccount.
+   */
   uint256 internal _activePurePremiums; // sum of pure-premiums of active policies - In Wad
+
+  /**
+   * @dev The surplus field keeps track of the surplus or deficit (when negative) of the actual payouts made by the
+   * PremiumsAccount versus the collected pure premiums. On the negative side, it has a limit defined by `_maxDeficit()`,
+   * after that limit, internal loans are taken from the eTokens.
+   */
   int256 internal _surplus;
 
   struct PackedParams {
@@ -41,14 +72,22 @@ contract PremiumsAccount is IPremiumsAccount, Reserve {
 
   PackedParams internal _params;
 
-  /*
-   * Premiums can come in (for free, without liability) with receiveGrant.
+  /**
+   * Premiums can come in (for "free", without liability) with receiveGrant.
    * And can come out (withdrawed to treasury) with withdrawWonPremiums
+   *
+   * @param moneyIn Indicates if money came in or out (false).
+   * @param value The amount of money received or given
    */
   event WonPremiumsInOut(bool moneyIn, uint256 value);
 
+  /**
+   * @dev Constructor of the contract, sets the immutable fields.
+   *
+   * @param juniorEtk_ Address of the Junior EToken (first loss lender). `address(0)` if not present.
+   * @param seniorEtk_ Address of the Senior EToken (2nd loss lender). `address(0)` if not present.
+   */
   /// @custom:oz-upgrades-unsafe-allow constructor
-  // solhint-disable-next-line no-empty-blocks
   constructor(
     IPolicyPool policyPool_,
     IEToken juniorEtk_,
@@ -59,35 +98,54 @@ contract PremiumsAccount is IPremiumsAccount, Reserve {
   }
 
   /**
-   * @dev Public initialize Initializes the PremiumsAccount
+   * @dev Initializes the PremiumsAccount
    */
   function initialize() public initializer {
     __PremiumsAccount_init();
   }
 
   /**
-   * @dev Initializes the PremiumsAccount
+   * @dev Initializes the PremiumsAccount (to be called by subclasses)
    */
   // solhint-disable-next-line func-name-mixedcase
-  function __PremiumsAccount_init() internal initializer {
-    __PolicyPoolComponent_init();
+  function __PremiumsAccount_init() internal onlyInitializing {
+    __Reserve_init();
     __PremiumsAccount_init_unchained();
   }
 
   // solhint-disable-next-line func-name-mixedcase
-  function __PremiumsAccount_init_unchained() internal initializer {
+  function __PremiumsAccount_init_unchained() internal onlyInitializing {
     /*
     _activePurePremiums = 0;
     */
-    if (address(_juniorEtk) != address(0))
-      currency().approve(address(_juniorEtk), type(uint256).max);
-    if (address(_seniorEtk) != address(0))
-      currency().approve(address(_seniorEtk), type(uint256).max);
-
-    _params = PackedParams({deficitRatio: 1e4, assetManager: IAssetManager(address(0))});
+    _params = PackedParams({
+      deficitRatio: HUNDRED_PERCENT,
+      assetManager: IAssetManager(address(0))
+    });
     _validateParameters();
   }
 
+  function _upgradeValidations(address newImpl) internal view virtual override {
+    super._upgradeValidations(newImpl);
+    IPremiumsAccount newPA = IPremiumsAccount(newImpl);
+    require(
+      newPA.juniorEtk() == _juniorEtk || address(_juniorEtk) == address(0),
+      "Can't upgrade changing the Junior ETK unless to non-zero"
+    );
+    require(
+      newPA.seniorEtk() == _seniorEtk || address(_seniorEtk) == address(0),
+      "Can't upgrade changing the Senior ETK unless to non-zero"
+    );
+  }
+
+  /**
+   * @dev See {IERC165-supportsInterface}.
+   */
+  function supportsInterface(bytes4 interfaceId) public view virtual override returns (bool) {
+    return
+      super.supportsInterface(interfaceId) || interfaceId == type(IPremiumsAccount).interfaceId;
+  }
+
   function assetManager() public view override returns (IAssetManager) {
     return _params.assetManager;
   }
@@ -96,6 +154,14 @@ contract PremiumsAccount is IPremiumsAccount, Reserve {
     _params.assetManager = newAM;
   }
 
+  /**
+   * @dev This is called by the {Reserve} base class to record the earnings generated by the asset management.
+   *
+   * @param earningsOrLosses Indicates the amount earned since last time earnings where recorded.
+   * - If positive, repays the loans and accumulates the rest in the surplus.
+   * - If negative (losses) substracts it from surplus. It never can exceed _maxDeficit and doesn't takes
+   *   loans to cover asset losses.
+   */
   function _assetEarnings(int256 earningsOrLosses) internal override {
     if (earningsOrLosses > 0) {
       uint256 earnings = uint256(earningsOrLosses);
@@ -103,14 +169,13 @@ contract PremiumsAccount is IPremiumsAccount, Reserve {
       if (address(_juniorEtk) != address(0)) earnings = _repayLoan(earnings, _juniorEtk);
       _storePurePremiumWon(earnings);
     } else {
-      _payFromPremiums(uint256(-earningsOrLosses));
+      require(_payFromPremiums(uint256(-earningsOrLosses)) == 0, "Losses can't exceed maxDeficit");
     }
   }
 
-  // solhint-disable-next-line no-empty-blocks
   function _validateParameters() internal view override {
     require(
-      _params.deficitRatio <= 1e4 && _params.deficitRatio >= 0,
+      _params.deficitRatio <= HUNDRED_PERCENT && _params.deficitRatio >= 0,
       "Validation: deficitRatio must be <= 1"
     );
   }
@@ -119,18 +184,35 @@ contract PremiumsAccount is IPremiumsAccount, Reserve {
     return uint256(int256(_activePurePremiums) + _surplus);
   }
 
+  /**
+   * @dev Returns the total amount of pure premiums that were collected by the active policies of the risk modules
+   * linked to this PremiumsAccount.
+   */
   function activePurePremiums() external view returns (uint256) {
     return _activePurePremiums;
   }
 
+  /**
+   * @dev Returns the surplus between pure premiums collected and payouts of finalized policies. Returns 0 if no surplus
+   * or deficit.
+   */
   function wonPurePremiums() external view returns (uint256) {
     return _surplus >= 0 ? uint256(_surplus) : 0;
   }
 
+  /**
+   * @dev Returns the amount of active pure premiums that was used to cover payouts of finalized policies (in excess of
+   * collected pure premiums). This is limited by `_maxDeficit()`
+   */
   function borrowedActivePP() external view returns (uint256) {
     return _surplus >= 0 ? 0 : uint256(-_surplus);
   }
 
+  /**
+   * @dev Returns the surplus between pure premiums collected and payouts of finalized policies. Losses where more than
+   * premiums collected, returns a negative number that indicates the amount of the active pure premiums that was used
+   * to cover finalized premiums.
+   */
   function surplus() external view returns (int256) {
     return _surplus;
   }
@@ -143,35 +225,83 @@ contract PremiumsAccount is IPremiumsAccount, Reserve {
     return _juniorEtk;
   }
 
+  /**
+   * @dev Returns the maximum deficit that's supported by the PremiumsAccount. If more money is needed, it must take
+   * loans from the eTokens. The value is calculated as a fraction of the active pure premiums. The fraction is
+   * regulated by the `deficitRatio` parameter that indicates the percentage of the active pure premiums that can be
+   * used to cover payouts of finalized policies. In many cases is fine to use the active pure premiums to cover the
+   * losses because in most cases the policies with payout are triggered long time before the policies without payout.
+   * But this also can be dangerous because it can be postponing the losses that should impact on liquidity providers.
+   *
+   * @param ratio The ratio used in the calculation of the deficit. It's the deficitRatio parameter (whether the current
+   * one or the new one when it's being modified).
+   */
   function _maxDeficit(uint256 ratio) internal view returns (int256) {
     return -int256(_activePurePremiums.wadMul(ratio));
   }
 
+  /**
+   * @dev Returns the percentage of the active pure premiums that can be used to cover losses of finalized policies.
+   */
   function deficitRatio() public view returns (uint256) {
-    return uint256(_params.deficitRatio) * 1e14; // 4 -> 18 decimals
+    return uint256(_params.deficitRatio) * FOUR_DECIMAL_TO_WAD; // 4 -> 18 decimals
   }
 
+  /**
+   * @dev Changes the `deficitRatio` parameter.
+   *
+   * Requirements:
+   * - onlyGlobalOrComponentRole(LEVEL2_ROLE)
+   *
+   * Events:
+   * - Emits GovernanceAction with action = setDeficitRatio or setDeficitRatioWithAdjustment if an adjustment was made.
+   *
+   * @param adjustment If true and the new ratio leaves `_surplus < -_maxDeficit()`, it adjusts the _surplus to the new
+   *                   `_maxDeficit()` and borrows the difference from the eTokens.
+   *                   If false and the new ratio leaves `_surplus < -_maxDeficit()`, the operation is reverted.
+   */
   function setDeficitRatio(uint256 newRatio, bool adjustment)
     external
     onlyComponentRole(LEVEL2_ROLE)
   {
-    require(newRatio <= 1e18 && newRatio >= 0, "Validation: deficitRatio must be <= 1");
+    require(newRatio <= 1e18, "Validation: deficitRatio must be <= 1");
+
+    uint16 truncatedRatio = (newRatio / FOUR_DECIMAL_TO_WAD).toUint16();
+    require(
+      uint256(truncatedRatio) * FOUR_DECIMAL_TO_WAD == newRatio,
+      "Validation: only up to 4 decimals allowed"
+    );
+
     int256 maxDeficit = _maxDeficit(newRatio);
     require(adjustment || _surplus >= maxDeficit, "Validation: surplus must be >= maxDeficit");
+
+    IAccessManager.GovernanceActions action = IAccessManager.GovernanceActions.setDeficitRatio;
     if (_surplus < maxDeficit) {
       // Do the adjustment
       uint256 borrow = uint256(-_surplus + maxDeficit);
       _surplus = maxDeficit;
       _borrowFromEtk(borrow, address(this), address(_juniorEtk) != address(0));
+      action = IAccessManager.GovernanceActions.setDeficitRatioWithAdjustment;
     }
-    _params.deficitRatio = uint16(newRatio / 1e14);
+    _params.deficitRatio = truncatedRatio;
+    _parameterChanged(action, newRatio, false);
   }
 
+  /**
+   * @dev Internal function called when money in the PremiumsAccount is not enough and we need to borrow from the
+   * eTokens.
+   *
+   * @param borrow The amount to borrow.
+   * @param receiver The address that will receive the money of the loan. Usually is the policy holder if this is called
+   *                 in the context of a policy payout.
+   * @param jrEtk If true it indicates that the loan is asked first from the junior eToken.
+   */
   function _borrowFromEtk(
     uint256 borrow,
     address receiver,
     bool jrEtk
   ) internal {
+    require(receiver != address(0), "PremiumsAccount: receiver cannot be the zero address");
     uint256 left;
     if (jrEtk) {
       // Consume Junior Pool until exhausted
@@ -186,6 +316,13 @@ contract PremiumsAccount is IPremiumsAccount, Reserve {
     }
   }
 
+  /**
+   * @dev Updates the `_surplus` field with the payment made. Since the _surplus can never exceed `_maxDeficit()`,
+   * returns the remaining amount in case something can't be paid from the PremiumsAccount.
+   *
+   * @param toPay The amount to pay.
+   * @return The amount that couldn't be paid from the premiums account.
+   */
   function _payFromPremiums(uint256 toPay) internal returns (uint256) {
     int256 newSurplus = _surplus - int256(toPay);
     int256 maxDeficit = _maxDeficit(deficitRatio());
@@ -197,44 +334,61 @@ contract PremiumsAccount is IPremiumsAccount, Reserve {
     return uint256(-newSurplus + maxDeficit);
   }
 
+  /**
+   * @dev Stores an earned pure premium. Adds to the surplus, increasing the surplus if it was positive or reducing the
+   * deficit if it was negative.
+   *
+   * @param purePremiumWon The amount earned
+   */
   function _storePurePremiumWon(uint256 purePremiumWon) internal {
     if (purePremiumWon == 0) return;
     _surplus += int256(purePremiumWon);
   }
 
-  // TODO: restore repayETokenLoan?
-
   /**
    *
    * Endpoint to receive "free money" and inject that money into the premium pool.
    *
    * Can be used for example if the PolicyPool subscribes an excess loss policy with other company.
    *
+   * Requirements:
+   * - The sender needs to approve the spending of `currency()` by this contract.
+   *
+   * Events:
+   * - Emits {WonPremiumsInOut} with moneyIn = true
+   *
+   * @param amount The amount to be transferred.
    */
   function receiveGrant(uint256 amount) external {
-    currency().safeTransferFrom(msg.sender, address(this), amount);
     _storePurePremiumWon(amount);
     emit WonPremiumsInOut(true, amount);
+    currency().safeTransferFrom(_msgSender(), address(this), amount);
   }
 
   /**
    *
-   * Withdraws excess premiums to PolicyPool's treasury.
+   * Withdraws excess premiums (surplus) to the destination.
+   *
    * This might be needed in some cases for example if we are deprecating the protocol or the excess premiums
    * are needed to compensate something. Shouldn't be used. Can be disabled revoking role WITHDRAW_WON_PREMIUMS_ROLE
    *
-   * returns The amount withdrawed
-   *
    * Requirements:
-   *
    * - onlyGlobalOrComponentRole(WITHDRAW_WON_PREMIUMS_ROLE)
-   * - _wonPurePremiums > 0
+   * - _surplus > 0
+   *
+   * Events:
+   * - Emits {WonPremiumsInOut} with moneyIn = false
+   *
+   * @param amount The amount to withdraw
+   * @param destination The address that will receive the transferred funds.
+   * @return Returns the actual amount withdrawn.
    */
   function withdrawWonPremiums(uint256 amount, address destination)
     external
     onlyGlobalOrComponentRole(WITHDRAW_WON_PREMIUMS_ROLE)
     returns (uint256)
   {
+    require(destination != address(0), "PremiumsAccount: destination cannot be the zero address");
     if (_surplus <= 0) {
       amount = 0;
     } else {
@@ -283,6 +437,11 @@ contract PremiumsAccount is IPremiumsAccount, Reserve {
     }
   }
 
+  /**
+   * @dev Internal function that calls the eTokens to lock the solvency capital when the policy is created.
+   *
+   * @param policy The policy created
+   */
   function _unlockScr(Policy.PolicyData memory policy) internal {
     if (policy.jrScr > 0) {
       _juniorEtk.unlockScr(
@@ -300,12 +459,34 @@ contract PremiumsAccount is IPremiumsAccount, Reserve {
     }
   }
 
+  /**
+   * @dev Internal function that repays a loan taken (if any outstanding) from the an eToken
+   *
+   * @param purePremiumWon The amount earned and available for loan repayment.
+   * @param etk The eToken with the potential debt
+   * @return The excess amount of the purePremiumWon that wasn't used for the loan repayment.
+   */
   function _repayLoan(uint256 purePremiumWon, IEToken etk) internal returns (uint256) {
     if (purePremiumWon < NEGLIGIBLE_AMOUNT) return purePremiumWon;
     uint256 borrowedFromEtk = etk.getLoan(address(this));
     if (borrowedFromEtk == 0) return purePremiumWon;
-    uint256 repayAmount = borrowedFromEtk > purePremiumWon ? purePremiumWon : borrowedFromEtk;
-    // TODO: make sure the balance is available or deinvest
+    uint256 repayAmount = Math.min(purePremiumWon, borrowedFromEtk);
+
+    // If not enough liquidity, it deinvests from the asset manager
+    if (currency().balanceOf(address(this)) < repayAmount) {
+      /**
+       * I send `repayAmount` because the IAssetManager expects the full amount that's needed, not the missing one.
+       * It uses the value of the full amount to optimize the deinvestment leaving more liquidity if possible to avoid
+       * future deinvestment. It will only fail if it can't refill `repayAmount - currency().balanceOf(address(this))`
+       */
+      _refillWallet(repayAmount);
+    }
+    // Checks the allowance before repayment
+    if (currency().allowance(address(this), address(etk)) < repayAmount) {
+      // If I have to approve, I approve for all the pending debt (not just repayAmount), this way I avoid some
+      // future approvals.
+      currency().approve(address(etk), borrowedFromEtk);
+    }
     etk.repayLoan(repayAmount, address(this));
     return purePremiumWon - repayAmount;
   }
@@ -334,4 +515,11 @@ contract PremiumsAccount is IPremiumsAccount, Reserve {
     _storePurePremiumWon(purePremiumWon);
     _unlockScr(policy);
   }
+
+  /**
+   * @dev This empty reserved space is put in place to allow future versions to add new
+   * variables without shifting down storage in the inheritance chain.
+   * See https://docs.openzeppelin.com/contracts/4.x/upgradeable#storage_gaps
+   */
+  uint256[47] private __gap;
 }