[fix] initialBalance to reset when flush

ycryptobank · Feb 20, 2024 · 3ae4243 · 3ae4243
1 parent 0265c58
commit 3ae4243
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/contracts/PasspostFinance-NFT.sol b/contracts/PasspostFinance-NFT.sol
@@ -315,6 +315,7 @@ contract YCBPassportFinance is
         if (erc20Address == address(sToken)) {
             // To avoid owner flush customer stake token
             contractBalance = initialBalance;
+            initialBalance = 0;
         }
 
         require(contractBalance > 0, "No tokens to flush");