chore: address review feedback

Signed-off-by: Tomás Migone <tomas@edgeandnode.com>

Author: tmigone

packages/contracts/contracts/rewards/RewardsManager.sol

@@ -361,7 +361,7 @@ contract RewardsManager is RewardsManagerV5Storage, GraphUpgradeable, IRewardsMa
     /**
      * @inheritdoc IRewardsManager
      * @dev This function can only be called by an authorized rewards issuer which are
-     * - the subgraph service (for new allocations).
+     * - the subgraph service (for allocations).
      * Mints 0 tokens if the allocation is not active.
      */
     function takeRewards(address _allocationID) external override returns (uint256) {

packages/horizon/contracts/staking/HorizonStaking.sol

@@ -545,9 +545,8 @@ contract HorizonStaking is HorizonStakingBase, IHorizonStakingMain {
         require(_tokens <= tokensIdle, HorizonStakingInsufficientIdleStake(_tokens, tokensIdle));
 
         ServiceProviderInternal storage sp = _serviceProviders[serviceProvider];
-        uint256 stakedTokens = sp.tokensStaked;
+        sp.tokensStaked -= _tokens;
 
-        sp.tokensStaked = stakedTokens - _tokens;
         _graphToken().pushTokens(serviceProvider, _tokens);
         emit HorizonStakeWithdrawn(serviceProvider, _tokens);
     }