File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -163,22 +163,22 @@ library SafeMath {
163
163
164
164
/* Contract class for adding removing whitelisted contracts */
165
165
contract WhiteListedContracts is Ownable {
166
- mapping (address => uint ) white_listed_contracts;
166
+ mapping (address => bool ) white_listed_contracts;
167
167
168
168
//modifer to check if the contract given is white listed or not
169
169
modifier whitelistedContractsOnly () {
170
- require (white_listed_contracts[msg .sender ] == 1 );
170
+ require (white_listed_contracts[msg .sender ]);
171
171
_;
172
172
}
173
173
174
174
//add a contract to whitelist
175
175
function addWhiteListedContracts (address _address ) onlyOwner public {
176
- white_listed_contracts[_address] = 1 ;
176
+ white_listed_contracts[_address] = true ;
177
177
}
178
178
179
179
//remove contract from whitelist
180
180
function removeWhiteListedContracts (address _address ) onlyOwner public {
181
- white_listed_contracts[_address] = 0 ;
181
+ white_listed_contracts[_address] = false ;
182
182
}
183
183
}
184
184
You can’t perform that action at this time.
0 commit comments