update bucket lends in remove collateral

ajna-finance · Aug 22, 2023 · 1ae859f · 1ae859f
1 parent 8768993
commit 1ae859f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/mappings/erc-721-pool.ts b/src/mappings/erc-721-pool.ts
@@ -354,10 +354,12 @@ export function handleRemoveCollateral(event: RemoveCollateralEvent): void {
   // splice the identified tokenIds out of pool.bucketTokenIds
   pool.bucketTokenIds = findAndRemoveTokenIds(tokenIdsToRemove, pool.bucketTokenIds)
 
-  // update account's list of pools and lends if necessary
+  // update bucket and account's list of pools and lends if necessary
+  updateBucketLends(bucket, lend)
   updateAccountPools(account, pool)
   updateAccountLends(account, lend)
-    // remove lend from store if necessary
+
+  // remove lend from store if necessary
   const isRemoved = removeLendFromStore(lend)
   if (!isRemoved) lend.save()