diff --git a/src/BlockBuilderPolicy.sol b/src/BlockBuilderPolicy.sol index d8c5226..a82022f 100644 --- a/src/BlockBuilderPolicy.sol +++ b/src/BlockBuilderPolicy.sol @@ -89,6 +89,7 @@ contract BlockBuilderPolicy is /// @inheritdoc IBlockBuilderPolicy function initialize(address _initialOwner, address _registry) external override initializer { __Ownable_init(_initialOwner); + __UUPSUpgradeable_init(); __EIP712_init("BlockBuilderPolicy", "1"); require(_registry != address(0), InvalidRegistry()); diff --git a/src/FlashtestationRegistry.sol b/src/FlashtestationRegistry.sol index 647a832..36d3635 100644 --- a/src/FlashtestationRegistry.sol +++ b/src/FlashtestationRegistry.sol @@ -66,6 +66,7 @@ contract FlashtestationRegistry is __Ownable_init(owner); __EIP712_init("FlashtestationRegistry", "1"); __ReentrancyGuardTransient_init(); + __UUPSUpgradeable_init(); require(_attestationContract != address(0), InvalidAttestationContract()); attestationContract = IAttestation(_attestationContract); }