Swapped stakedIndexersCount subtraction from withdraw to locked

graphprotocol · Apr 8, 2021 · e18fe9b · e18fe9b
1 parent 7b3e9b8
commit e18fe9b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/mappings/staking.ts b/src/mappings/staking.ts
@@ -108,6 +108,9 @@ export function handleStakeLocked(event: StakeLocked): void {
   graphNetwork.totalUnstakedTokensLocked = graphNetwork.totalUnstakedTokensLocked.plus(
     event.params.tokens,
   )
+  if (indexer.stakedTokens == indexer.lockedTokens) {
+    graphNetwork.stakedIndexersCount = graphNetwork.stakedIndexersCount - 1
+  }
   graphNetwork.save()
 }
 
@@ -133,9 +136,6 @@ export function handleStakeWithdrawn(event: StakeWithdrawn): void {
   graphNetwork.totalUnstakedTokensLocked = graphNetwork.totalUnstakedTokensLocked.minus(
     event.params.tokens,
   )
-  if (indexer.stakedTokens == BigInt.fromI32(0)) {
-    graphNetwork.stakedIndexersCount = graphNetwork.stakedIndexersCount - 1
-  }
   graphNetwork.save()
 }