From 32e49edceb422140c159f709b8715a015d120298 Mon Sep 17 00:00:00 2001
From: wcgcyx <wcgcyx@gmail.com>
Date: Wed, 21 Feb 2024 14:20:36 +1000
Subject: [PATCH] Update InvariantBridge.t.sol

---
 test/invariant/InvariantBridge.t.sol | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/test/invariant/InvariantBridge.t.sol b/test/invariant/InvariantBridge.t.sol
index c9c645df..650a47d4 100644
--- a/test/invariant/InvariantBridge.t.sol
+++ b/test/invariant/InvariantBridge.t.sol
@@ -215,7 +215,14 @@ contract InvariantBridge is Test {
             vm.selectFork(childId);
             uint256 totalSupply = ChildERC20(childToken).totalSupply();
 
+            uint256 userBalanceSum = 0;
+            for (uint256 j = 0; j < NO_OF_USERS; j++) {
+                address user = users[j];
+                userBalanceSum += ChildERC20(childToken).balanceOf(user);
+            }
+
             assertEq(bridgeBalance, totalSupply);
+            assertEq(bridgeBalance, userBalanceSum);
         }
     }
 }