From aa378e6ef58f1cab87e1f03e2b40cd72d51ed152 Mon Sep 17 00:00:00 2001
From: Louise Poole <louisecarmenpoole@gmail.com>
Date: Fri, 21 Mar 2025 11:34:37 +0200
Subject: [PATCH] fix: handle token balances on add_contract_changes (#181)

---
 substreams/crates/tycho-substreams/src/models.rs | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/substreams/crates/tycho-substreams/src/models.rs b/substreams/crates/tycho-substreams/src/models.rs
index e41cfeb..a99949c 100644
--- a/substreams/crates/tycho-substreams/src/models.rs
+++ b/substreams/crates/tycho-substreams/src/models.rs
@@ -52,6 +52,9 @@ impl TransactionChangesBuilder {
                 if !change.code.is_empty() {
                     c.set_code(&change.code)
                 }
+                if !change.token_balances.is_empty() {
+                    c.upsert_token_balances(&change.token_balances)
+                }
             })
             .or_insert_with(|| {
                 let mut c = InterimContractChange::new(
@@ -61,6 +64,7 @@ impl TransactionChangesBuilder {
                 c.upsert_slots(&change.slots);
                 c.set_code(&change.code);
                 c.set_balance(&change.balance);
+                c.upsert_token_balances(&change.token_balances);
                 c
             });
     }