fix: use onlyowner

VenusProtocol · Aug 25, 2023 · 52f8efd · 52f8efd
1 parent 1992f80
commit 52f8efd
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/contracts/Admin/VBNBAdmin.sol b/contracts/Admin/VBNBAdmin.sol
@@ -84,8 +84,7 @@ contract VBNBAdmin is ReentrancyGuardUpgradeable, AccessControlledV8, VBNBAdminS
     /**
      * @notice Accept admin for vBNB
      */
-    function acceptVBNBAdmin() external nonReentrant returns (uint) {
-        require(msg.sender == owner(), "only owner can accept admin");
+    function acceptVBNBAdmin() external onlyOwner nonReentrant returns (uint) {
         return vBNB._acceptAdmin();
     }
 
@@ -107,9 +106,7 @@ contract VBNBAdmin is ReentrancyGuardUpgradeable, AccessControlledV8, VBNBAdminS
     /**
      * @notice Invoked when called function does not exist in the contract
      */
-    fallback(bytes calldata data) external payable returns (bytes memory) {
-        require(msg.sender == owner(), "only owner can call vBNB admin functions");
-
+    fallback(bytes calldata data) external payable onlyOwner returns (bytes memory) {
         (bool ok, bytes memory res) = address(vBNB).call{ value: msg.value }(data);
         require(ok, "call failed");
         return res;