Override _totalValueInVault

OriginProtocol · Jul 15, 2024 · ef45537 · ef45537
1 parent cbd6764
commit ef45537
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/contracts/contracts/vault/OETHVaultCore.sol b/contracts/contracts/vault/OETHVaultCore.sol
@@ -375,8 +375,15 @@ contract OETHVaultCore is VaultCore {
         super._allocate();
     }
 
-    function _totalValue() internal view override returns (uint256 value) {
-        value = super._totalValue();
+    function _totalValueInVault()
+        internal
+        view
+        virtual
+        override
+        returns (uint256 value)
+    {
+        // Only WETH is counted
+        value = IERC20(weth).balanceOf(address(this));
 
         // Need to remove WETH that is reserved for the withdrawal queue.
         // reserved for the withdrawal queue = cumulative queued total - total claimed