diff --git a/contracts/yield-source/ATokenYieldSource.sol b/contracts/yield-source/ATokenYieldSource.sol index ac3fb48..1ef67e9 100644 --- a/contracts/yield-source/ATokenYieldSource.sol +++ b/contracts/yield-source/ATokenYieldSource.sol @@ -99,6 +99,8 @@ contract ATokenYieldSource is ERC20Upgradeable, IProtocolYieldSource, AssetManag transferOwnership(_owner); __ERC20_init(_name,_symbol); + __ReentrancyGuard_init(); + require(_decimals > 0, "ATokenYieldSource/decimals-gt-zero"); _setupDecimals(_decimals);