minor dispenseLock refactor (#55)

The-Poolz · Dec 19, 2024 · 967aed8 · 967aed8
1 parent 0d191d0
commit 967aed8
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/contracts/DispenserInternal.sol b/contracts/DispenserInternal.sol
@@ -53,7 +53,6 @@ abstract contract DispenserInternal is DispenserState {
             if (data[i].params.length == 0) revert ZeroParamsLength();
             amountTaken += _nftIterator(tokenPoolId, receiver, data[i]);
         }
-        _finalizeDeal(tokenPoolId, receiver, amountTaken);
     }
 
     /// @notice Iterates through the NFTs and dispenses them from the pool.

diff --git a/contracts/DispenserProvider.sol b/contracts/DispenserProvider.sol
@@ -41,6 +41,7 @@ contract DispenserProvider is DispenserModifiers {
         isValidSignature(poolId, validUntil, receiver, data, signature)
     {
         uint256 amountTaken = _handleSimpleNFTs(poolId, receiver, data);
+        _finalizeDeal(poolId, receiver, amountTaken);
 
         emit TokensDispensed(
             poolId,