diff --git a/contracts/mocks/EnumerableSetMock.sol b/contracts/mocks/EnumerableSetMock.sol index 10c4470de..e4456fba5 100644 --- a/contracts/mocks/EnumerableSetMock.sol +++ b/contracts/mocks/EnumerableSetMock.sol @@ -14,7 +14,7 @@ contract EnumerableSetMock{ EnumerableSet.AddressSet private set; constructor() public { - set = EnumerableSet.AddressSet({values: new address[](0)}); + set = EnumerableSet.newAddressSet(); } /** diff --git a/contracts/utils/EnumerableSet.sol b/contracts/utils/EnumerableSet.sol index 292ed7078..a8fb56863 100644 --- a/contracts/utils/EnumerableSet.sol +++ b/contracts/utils/EnumerableSet.sol @@ -27,6 +27,17 @@ library EnumerableSet { address[] values; } + /** + * @dev Add a value. O(1). + */ + function newAddressSet() + internal + pure + returns (AddressSet memory) + { + return AddressSet({values: new address[](0)}); + } + /** * @dev Add a value. O(1). */