fix: Whitelist onlyInitializing

thesandboxgame · Oct 6, 2023 · 9dcdf81 · 9dcdf81 · github-actions · Oct 6, 2023
1 parent de73e19
commit 9dcdf81
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/packages/marketplace/contracts/exchange/WhiteList.sol b/packages/marketplace/contracts/exchange/WhiteList.sol
@@ -2,12 +2,13 @@
 
 pragma solidity 0.8.21;
 
+import {Initializable} from "@openzeppelin/contracts-upgradeable/proxy/utils/Initializable.sol";
 import {AccessControlEnumerableUpgradeable} from "@openzeppelin/contracts-upgradeable/access/AccessControlEnumerableUpgradeable.sol";
 import {IWhiteList} from "../interfaces/IWhiteList.sol";
 
 /// @title WhiteList contract
 /// @dev controls which tokens are accepted in the marketplace
-contract WhiteList is IWhiteList, AccessControlEnumerableUpgradeable {
+contract WhiteList is Initializable, IWhiteList, AccessControlEnumerableUpgradeable {
     /// @notice role for The Sandbox tokens
     /// @return hash for TSB_ROLE
     bytes32 public constant TSB_ROLE = keccak256("TSB_ROLE");
@@ -60,7 +61,7 @@ contract WhiteList is IWhiteList, AccessControlEnumerableUpgradeable {
         bool newPartners,
         bool newOpen,
         bool newErc20List
-    ) internal initializer {
+    ) internal onlyInitializing {
         __AccessControlEnumerable_init_unchained();
         _grantRole(DEFAULT_ADMIN_ROLE, admin);
         _setPermissions(newTsbOnly, newPartners, newOpen, newErc20List);