fix: address and added handlers

peterferguson · Mar 17, 2024 · 8ae4a40 · 8ae4a40
1 parent 61d63d4
commit 8ae4a40
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/packages/contracts/script/deploy/Safe4337FactoryDeployer.s.sol b/packages/contracts/script/deploy/Safe4337FactoryDeployer.s.sol
@@ -11,7 +11,7 @@ contract Safe4337FactoryDeployer is Script {
     // v1.4.1
     address public constant SAFE_SINGLETON_ADDRESS = 0x29fcB43b46531BcA003ddC8FCB67FFE91900C762;
     // Deployed in other script
-    address public constant ADD_MODULES_LIB_ADDRESS = 0x4e1DCf7AD4e460CfD30791CCC4F9c8a4f820ec67;
+    address public constant ADD_MODULES_LIB_ADDRESS = 0x58E912c126F92cCD3C6856a0d1104a30D5260E2b;
     // v0.7.0
     address public constant ENTRY_POINT_ADDRESS = 0x0000000071727De22E5E9d8BAf0edAc6f37da032;
 

diff --git a/packages/contracts/src/safe-4337-module/Safe4337Module.sol b/packages/contracts/src/safe-4337-module/Safe4337Module.sol
@@ -5,6 +5,8 @@ import {BaseAccount, IEntryPoint} from "../../lib/account-abstraction/contracts/
 import {PackedUserOperation} from "../../lib/account-abstraction/contracts/interfaces/PackedUserOperation.sol";
 import {HandlerContext} from "../../lib/safe-smart-account/contracts/handler/HandlerContext.sol";
 import {WebAuthn} from "../../lib/webauthn-sol/src/WebAuthn.sol";
+import {CompatibilityFallbackHandler} from
+    "../../lib/safe-smart-account/contracts/handler/CompatibilityFallbackHandler.sol";
 
 import {ISafe} from "../interfaces/ISafe.sol";
 
@@ -21,7 +23,7 @@ import "forge-std/console.sol";
 /**
  * TODO
  */
-contract Safe4337Module is BaseAccount, HandlerContext {
+contract Safe4337Module is BaseAccount, HandlerContext, CompatibilityFallbackHandler {
     /// ----------------------------------------------------------------------------------------
     ///							ACCOUNT STORAGE
     /// ----------------------------------------------------------------------------------------