From 45e36f8e0ea26e836c508fe1d9fd2385f6e31aa3 Mon Sep 17 00:00:00 2001
From: CounterFire2023 <136581895+CounterFire2023@users.noreply.github.com>
Date: Tue, 18 Jun 2024 14:18:52 +0800
Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E6=95=B4=E7=8A=B6=E6=80=81=E6=95=B0?=
 =?UTF-8?q?=E6=8D=AE=E5=86=99=E5=85=A5=E6=97=B6=E6=9C=BA,=20=E5=B7=B2?=
 =?UTF-8?q?=E9=98=B2=E6=AD=A2cross=20function=20reentrancy?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 contracts/activity/NFTClaimStage2WL.sol | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/contracts/activity/NFTClaimStage2WL.sol b/contracts/activity/NFTClaimStage2WL.sol
index 56bdad3..44eec3b 100644
--- a/contracts/activity/NFTClaimStage2WL.sol
+++ b/contracts/activity/NFTClaimStage2WL.sol
@@ -113,6 +113,7 @@ contract NFTClaimStage2WL is ReentrancyGuard, AccessControl {
       require(_whitelist1[to] + _whitelist2[to] >= _mintedCount + nftCount, "NFTClaimer: not in whitelist or exceed limit");
     }
     uint256 _tokenAmount = mintConfig.mintPrice * nftCount;
+    totalCount += nftCount;
     IERC20(mintConfig.currency).safeTransferFrom(to, mintConfig.feeToAddress, _tokenAmount);
     uint256[] memory ids = new uint256[](nftCount);
     for (uint256 i = 0; i < nftCount; ++i) {
@@ -121,7 +122,6 @@ contract NFTClaimStage2WL is ReentrancyGuard, AccessControl {
       IClaimAbleNFT(nftAddress).safeMint(to, _nftId);
       _mintedRecords[to].add(_nftId);
     }
-    totalCount += nftCount;
     // add list
     emit NFTClaimed(nftAddress, to, ids);
   }