fix: Add check for already initialized guardian

jpgonzalezra · May 20, 2024 · b8e2cbd · b8e2cbd
1 parent 89967b4
commit b8e2cbd
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/SimpleGuardianModule.sol b/src/SimpleGuardianModule.sol
@@ -18,6 +18,7 @@ abstract contract SimpleGuardianModule {
     error InvalidGuardianSignature();
     error NotGuardian();
     error InvalidNonce();
+    error GuardianAlreadyInitialized();
 
     address public guardian;
     mapping(address => uint256) private _nonces;
@@ -37,7 +38,9 @@ abstract contract SimpleGuardianModule {
     }
 
     function initGuardian(address newGuardian) external {
-        require(guardian == address(0));
+        if (guardian != address(0)) {
+            revert GuardianAlreadyInitialized();
+        }
         _updateGuardian(newGuardian);
     }