From 98b27e7fb07126c8215519e8721657e51e9724c3 Mon Sep 17 00:00:00 2001
From: Pablo Veyrat <veyratpablo@gmail.com>
Date: Fri, 10 May 2024 13:36:23 +0200
Subject: [PATCH] fix: tests

---
 test/foundry/RouterSidechainSavingsRateActions.t.sol | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/test/foundry/RouterSidechainSavingsRateActions.t.sol b/test/foundry/RouterSidechainSavingsRateActions.t.sol
index 8c621a2..b558712 100644
--- a/test/foundry/RouterSidechainSavingsRateActions.t.sol
+++ b/test/foundry/RouterSidechainSavingsRateActions.t.sol
@@ -67,7 +67,7 @@ contract RouterSidechainSavingsRateActionsTest is BaseTest {
         address to,
         uint256 gainOrLoss
     ) public {
-        vm.assume(to != address(0) && to != address(savingsRate));
+        vm.assume(to != address(0) && to != address(savingsRate) && to != address(router));
         uint256 balanceUsers = BASE_TOKENS * 1 ether;
         deal(address(token), address(_alice), balanceUsers);
 
@@ -119,7 +119,7 @@ contract RouterSidechainSavingsRateActionsTest is BaseTest {
         address to,
         uint256 gainOrLoss
     ) public {
-        vm.assume(to != address(0) && to != address(savingsRate));
+        vm.assume(to != address(0) && to != address(savingsRate) && to != address(router));
         uint256 balanceUsers = BASE_TOKENS * 1 ether;
         deal(address(token), address(_alice), balanceUsers);
 
@@ -170,7 +170,7 @@ contract RouterSidechainSavingsRateActionsTest is BaseTest {
         uint256 gainOrLoss,
         uint256 gainOrLoss2
     ) public {
-        vm.assume(to != address(0) && to != address(savingsRate));
+        vm.assume(to != address(0) && to != address(savingsRate) && to != address(router));
         uint256 balanceUsers = BASE_TOKENS * 1 ether;
         deal(address(token), address(_alice), balanceUsers);
 
@@ -248,7 +248,7 @@ contract RouterSidechainSavingsRateActionsTest is BaseTest {
         uint256 gainOrLoss,
         uint256 gainOrLoss2
     ) public {
-        vm.assume(to != address(0) && to != address(savingsRate));
+        vm.assume(to != address(0) && to != address(savingsRate) && to != address(router));
         uint256 balanceUsers = BASE_TOKENS * 1 ether;
         deal(address(token), address(_alice), balanceUsers);