L01 - Ownable contract has a single step ownership transferred

DePayFi · Aug 29, 2023 · 00038cf · 00038cf
1 parent 7616863
commit 00038cf
Show file tree

Hide file tree

Showing 7 changed files with 822 additions and 298 deletions.
diff --git a/contracts/DePayForwarderV2.sol b/contracts/DePayForwarderV2.sol
@@ -2,11 +2,11 @@
 
 pragma solidity 0.8.18;
 
-import "@openzeppelin/contracts/access/Ownable.sol";
+import "@openzeppelin/contracts/access/Ownable2Step.sol";
 import "@openzeppelin/contracts/token/ERC20/utils/SafeERC20.sol";
 import './interfaces/IDePayRouterV2.sol';
 
-contract DePayForwarderV2 is Ownable {
+contract DePayForwarderV2 is Ownable2Step {
 
   using SafeERC20 for IERC20;
 

diff --git a/contracts/DePayRouterV2.sol b/contracts/DePayRouterV2.sol
@@ -2,13 +2,13 @@
 
 pragma solidity 0.8.18;
 
-import "@openzeppelin/contracts/access/Ownable.sol";
+import "@openzeppelin/contracts/access/Ownable2Step.sol";
 import "@openzeppelin/contracts/token/ERC20/utils/SafeERC20.sol";
 import './interfaces/IPermit2.sol';
 import './interfaces/IDePayRouterV2.sol';
 import './interfaces/IDePayForwarderV2.sol';
 
-contract DePayRouterV2 is Ownable {
+contract DePayRouterV2 is Ownable2Step {
 
   using SafeERC20 for IERC20;