Update KaliBerger.sol

kalidao · Nov 3, 2023 · f7707c4 · f7707c4
1 parent c59a028
commit f7707c4
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/src/KaliBerger.sol b/src/KaliBerger.sol
@@ -331,14 +331,11 @@ contract KaliBerger is Storage {
         collectPatronage(token, tokenId)
         onlyOwner(token, tokenId)
     {
-        if (this.getDeposit(token, tokenId) >= amount) {
-            (bool success,) = msg.sender.call{value: amount}("");
-            if (!success) revert TransferFailed();
+        if (amount > this.getDeposit(token, tokenId)) revert InvalidExit();
+        (bool success,) = msg.sender.call{value: amount}("");
+        if (!success) addUnclaimed(msg.sender, amount);
 
-            subDeposit(token, tokenId, amount);
-        } else {
-            revert InvalidExit();
-        }
+        subDeposit(token, tokenId, amount);
     }
 
     /// -----------------------------------------------------------------------