keep the modifier

bcnmy · Sep 4, 2024 · a7e1c10 · a7e1c10
1 parent 93c8e0d
commit a7e1c10
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/contracts/Nexus.sol b/contracts/Nexus.sol
@@ -313,7 +313,7 @@ contract Nexus is INexus, BaseAccount, ExecutionHelper, ModuleManager, UUPSUpgra
     /// as Biconomy v2 Account (proxy) reads implementation from the slot that is defined by its address
     /// @param newImplementation The address of the new contract implementation.
     /// @param data The calldata to be sent to the new implementation.
-    function upgradeToAndCall(address newImplementation, bytes calldata data) public payable virtual override withHook {
+    function upgradeToAndCall(address newImplementation, bytes calldata data) public payable virtual override onlyEntryPointOrSelf withHook {
         if (newImplementation == address(0)) {
             revert InvalidImplementationAddress();
         }