Merge pull request #310 from invariant-labs/fix-validate-state

fix validate state
invariant-labs · Oct 16, 2024 · b405362 · b405362
2 parents eba19c8 + 71f9abc
commit b405362
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cron/src/validate-state.ts b/cron/src/validate-state.ts
@@ -41,8 +41,8 @@ const main = async () => {
   for (const [poolKey, pool, liquidityTicks] of poolsWithTicks) {
     const { liquidityX, liquidityY } = getPairLiquidityValues(pool, liquidityTicks)
 
-    const newBalanceX = (balances.get(poolKey.tokenX) ?? 0n) + liquidityX
-    const newBalanceY = (balances.get(poolKey.tokenY) ?? 0n) + liquidityY
+    const newBalanceX = (balances.get(poolKey.tokenX) ?? 0n) + liquidityX + pool.feeProtocolTokenX
+    const newBalanceY = (balances.get(poolKey.tokenY) ?? 0n) + liquidityY + pool.feeProtocolTokenY
 
     balances.set(poolKey.tokenX, newBalanceX)
     balances.set(poolKey.tokenY, newBalanceY)