From b5b7f2d0065658d08678517d44c2e962b47e20e2 Mon Sep 17 00:00:00 2001
From: zizou <111426680+flopell@users.noreply.github.com>
Date: Tue, 10 Dec 2024 15:36:03 +0100
Subject: [PATCH] fix: correctly overwrites balances

We were overwriting balances only for sell and buy tokens. But some SwapAdapter rely on every balances of the pool being there which might not be the case if the pool has 2+ tokens.
---
 src/evm/protocol/vm/state.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/evm/protocol/vm/state.rs b/src/evm/protocol/vm/state.rs
index efcfc9a8..dc17aedf 100644
--- a/src/evm/protocol/vm/state.rs
+++ b/src/evm/protocol/vm/state.rs
@@ -219,13 +219,13 @@ where
         tokens: Vec<Address>,
         max_amount: U256,
     ) -> Result<HashMap<Address, Overwrites>, SimulationError> {
-        let sell_token = &tokens[0].clone();
+        let sell_token = &tokens[0].clone(); //TODO: need to make it clearer from the interface
         let mut res: Vec<HashMap<Address, Overwrites>> = Vec::new();
         if !self
             .capabilities
             .contains(&Capability::TokenBalanceIndependent)
         {
-            res.push(self.get_balance_overwrites(tokens)?);
+            res.push(self.get_balance_overwrites(self.tokens.clone())?);
         }
 
         let (slots, compiler) = self