diff --git a/src/FactoryTokenContractV2.sol b/src/FactoryTokenContractV2.sol index ee3eac1..1dd1395 100644 --- a/src/FactoryTokenContractV2.sol +++ b/src/FactoryTokenContractV2.sol @@ -457,4 +457,17 @@ contract FactoryTokenContractV2 is Ownable, ReentrancyGuard, Pausable { liquidityManager = LiquidityManager(_newManager); emit ContractUpgraded("LiquidityManager", oldManager, _newManager); } + + /** + * @notice Update vesting contract + * @param _newVesting New vesting contract address + */ + function updateVestingContract(address _newVesting) external onlyOwner { + if (_newVesting == address(0)) { + revert FactoryTokenContract__InvalidAddress(); + } + address oldVesting = address(vestingContract); + vestingContract = VestingContract(_newVesting); + emit ContractUpgraded("VestingContract", oldVesting, _newVesting); + } }