From 04fcd0aa83454bc0935ce9c047368dd4879a81f1 Mon Sep 17 00:00:00 2001
From: Miguel de Elias <miguel.deelias@gmail.com>
Date: Thu, 12 Dec 2024 17:14:58 -0300
Subject: [PATCH] fix: underflow in getIdleStake (TRST-L14)

---
 packages/horizon/contracts/staking/HorizonStakingBase.sol | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/packages/horizon/contracts/staking/HorizonStakingBase.sol b/packages/horizon/contracts/staking/HorizonStakingBase.sol
index 6745fb2e1..5f2808255 100644
--- a/packages/horizon/contracts/staking/HorizonStakingBase.sol
+++ b/packages/horizon/contracts/staking/HorizonStakingBase.sol
@@ -268,11 +268,11 @@ abstract contract HorizonStakingBase is
      * TODO: update the calculation after the transition period.
      */
     function _getIdleStake(address _serviceProvider) internal view returns (uint256) {
-        return
-            _serviceProviders[_serviceProvider].tokensStaked -
-            _serviceProviders[_serviceProvider].tokensProvisioned -
-            _serviceProviders[_serviceProvider].__DEPRECATED_tokensAllocated -
+        uint256 tokensUsed = _serviceProviders[_serviceProvider].tokensProvisioned +
+            _serviceProviders[_serviceProvider].__DEPRECATED_tokensAllocated +
             _serviceProviders[_serviceProvider].__DEPRECATED_tokensLocked;
+        uint256 tokensStaked = _serviceProviders[_serviceProvider].tokensStaked;
+        return tokensStaked > tokensUsed ? tokensStaked - tokensUsed : 0;
     }
 
     /**