diff --git a/nest/src/AggregateToken.sol b/nest/src/AggregateToken.sol index 4baffab..2ac2144 100644 --- a/nest/src/AggregateToken.sol +++ b/nest/src/AggregateToken.sol @@ -201,6 +201,10 @@ contract AggregateToken is ComponentToken, IAggregateToken, ERC1155Holder { IComponentToken componentToken, uint256 amount ) external nonReentrant onlyRole(ADMIN_ROLE) { + // Verify the componentToken is in componentTokenMap + if (!_getAggregateTokenStorage().componentTokenMap[componentToken]) { + revert ComponentTokenNotListed(componentToken); + } IERC20(componentToken.asset()).approve(address(componentToken), amount); }