diff --git a/src/RumpelWalletFactory.sol b/src/RumpelWalletFactory.sol index 468c48a..d127de5 100644 --- a/src/RumpelWalletFactory.sol +++ b/src/RumpelWalletFactory.sol @@ -88,6 +88,7 @@ contract RumpelWalletFactory is Ownable, Pausable { else if (what == "RUMPEL_MODULE") rumpelModule = data; else if (what == "RUMPEL_GUARD") rumpelGuard = data; else if (what == "INITIALIZATION_SCRIPT") initializationScript = data; + else if (what == "COMPATIBILITY_FALLBACK") compatibilityFallback = data; else revert UnrecognizedParam(what); emit ParamChanged(what, data); } diff --git a/test/RumpelWallet.t.sol b/test/RumpelWallet.t.sol index 9a8b684..af0be03 100644 --- a/test/RumpelWallet.t.sol +++ b/test/RumpelWallet.t.sol @@ -90,6 +90,7 @@ contract RumpelWalletTest is Test { address newScript = makeAddr("newScript"); address newSingleton = makeAddr("newSingleton"); address newProxyFactory = makeAddr("newProxyFactory"); + address newCompatibilityFallback = makeAddr("newCompatibilityFallback"); vm.startPrank(admin); rumpelWalletFactory.setParam("RUMPEL_GUARD", newGuard); @@ -97,6 +98,7 @@ contract RumpelWalletTest is Test { rumpelWalletFactory.setParam("INITIALIZATION_SCRIPT", newScript); rumpelWalletFactory.setParam("SAFE_SINGLETON", newSingleton); rumpelWalletFactory.setParam("PROXY_FACTORY", newProxyFactory); + rumpelWalletFactory.setParam("COMPATIBILITY_FALLBACK", newCompatibilityFallback); vm.stopPrank(); assertEq(rumpelWalletFactory.rumpelGuard(), newGuard); @@ -104,6 +106,7 @@ contract RumpelWalletTest is Test { assertEq(rumpelWalletFactory.initializationScript(), newScript); assertEq(rumpelWalletFactory.safeSingleton(), newSingleton); assertEq(address(rumpelWalletFactory.proxyFactory()), newProxyFactory); + assertEq(rumpelWalletFactory.compatibilityFallback(), newCompatibilityFallback); } function testFuzz_createWalletOwners(uint256 ownersLength, uint256 threshold) public {