fix more pr comments

ensuro · Nov 28, 2023 · 1c2aab8 · 1c2aab8
1 parent a9c6d57
commit 1c2aab8
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/contracts/SharedSmartVault.sol b/contracts/SharedSmartVault.sol
@@ -127,12 +127,10 @@ contract SharedSmartVault is AccessControlUpgradeable, UUPSUpgradeable, ERC4626U
     uint256 assets,
     uint256 shares
   ) internal virtual override {
+    uint256 prevBalance = _balance();
     super._deposit(caller, receiver, assets, shares);
     _collector.call(asset(), assets);
-    require(
-      _balance() == 0,
-      "SharedSmartVault: balance of the shared smart vault should be 0 after deposit"
-    );
+    require(_balance() == prevBalance, "SharedSmartVault: the balance should be the same.");
   }
 
   function maxDeposit(address owner) public view virtual override returns (uint256) {