From dba8fa1af009aadd03d7442cec2527cd95001931 Mon Sep 17 00:00:00 2001
From: Kiro <1143020035@qq.com>
Date: Mon, 29 Dec 2025 11:09:53 +0800
Subject: [PATCH] update vault contract

---
 contracts/ytVault/YTAssetFactory.sol | 2 +-
 contracts/ytVault/YTAssetVault.sol   | 5 +----
 2 files changed, 2 insertions(+), 5 deletions(-)

diff --git a/contracts/ytVault/YTAssetFactory.sol b/contracts/ytVault/YTAssetFactory.sol
index cdae3e4..a6a2bc5 100644
--- a/contracts/ytVault/YTAssetFactory.sol
+++ b/contracts/ytVault/YTAssetFactory.sol
@@ -113,7 +113,7 @@ contract YTAssetFactory is Initializable, UUPSUpgradeable, OwnableUpgradeable {
         uint256[] memory _redemptionTimes,
         uint256[] memory _initialYtPrices,
         address _usdcPriceFeed
-    ) external returns (address[] memory vaults) {
+    ) external onlyOwner returns (address[] memory vaults) {
         require(
             _names.length == _symbols.length &&
             _names.length == _managers.length &&
diff --git a/contracts/ytVault/YTAssetVault.sol b/contracts/ytVault/YTAssetVault.sol
index 28cd8cf..db5af69 100644
--- a/contracts/ytVault/YTAssetVault.sol
+++ b/contracts/ytVault/YTAssetVault.sol
@@ -250,13 +250,10 @@ contract YTAssetVault is
     function processBatchWithdrawals(uint256 _batchSize) 
         external 
         nonReentrant 
+        onlyManager
         whenNotPaused
         returns (uint256 processedCount, uint256 totalDistributed) 
     {
-        if (msg.sender != manager && msg.sender != factory) {
-            revert Forbidden();
-        }
-        
         if (_batchSize == 0) revert InvalidBatchSize();
         
         uint256 availableUSDC = IERC20(usdcAddress).balanceOf(address(this));