@@ -118,7 +118,10 @@ contract Registry is Context, AdminRole {
118118 ) external onlyAdmin {
119119 require (_adrs.length == _cnt, "Invalid number of addresses " );
120120 require (_trusts.length == _cnt, "Invalid number of trust values " );
121- require (_pendingBalances.length == _cnt, "Invalid number of pending balance values " );
121+ require (
122+ _pendingBalances.length == _cnt,
123+ "Invalid number of pending balance values "
124+ );
122125
123126 for (uint256 i = 0 ; i < _cnt; i++ ) {
124127 _register (_adrs[i], _trusts[i], _pendingBalances[i]);
@@ -285,7 +288,7 @@ contract Registry is Context, AdminRole {
285288
286289 function _remove (address _adr ) internal {
287290 require (_adr != address (0 ), "Cannot remove zero address " );
288- require (maxTrusts[_adr] != 0 , "Address is not a contributor " );
291+ require (EnumerableSet. contains (accounts, _adr) , "Address is not a contributor " );
289292
290293 EnumerableSet.remove (accounts, _adr);
291294 delete maxTrusts[_adr];
@@ -301,7 +304,7 @@ contract Registry is Context, AdminRole {
301304 _adr != address (0 ),
302305 "Cannot set pending balance for zero balance "
303306 );
304- require (maxTrusts[_adr] != 0 , "Address is not a contributor " );
307+ require (EnumerableSet. contains (accounts, _adr) , "Address is not a contributor " );
305308 require (
306309 cstkToken.balanceOf (_adr) == 0 ,
307310 "User has activated his membership "
@@ -317,7 +320,7 @@ contract Registry is Context, AdminRole {
317320 _adr != address (0 ),
318321 "Cannot set pending balance for zero balance "
319322 );
320- require (maxTrusts[_adr] != 0 , "Address is not a contributor " );
323+ require (EnumerableSet. contains (accounts, _adr) , "Address is not a contributor " );
321324 require (
322325 cstkToken.balanceOf (_adr) == 0 ,
323326 "User has activated his membership "
0 commit comments