diff --git a/contracts/src/arbitration/KlerosCore.sol b/contracts/src/arbitration/KlerosCore.sol
index c9793d089..ff916577a 100644
--- a/contracts/src/arbitration/KlerosCore.sol
+++ b/contracts/src/arbitration/KlerosCore.sol
@@ -525,7 +525,7 @@ contract KlerosCore is IArbitratorV2 {
     function setStakeBySortitionModule(
         address _account,
         uint96 _courtID,
-        uint256 _stake,
+        uint256 _newStake,
         bool _alreadyTransferred
     ) external {
         if (msg.sender != address(sortitionModule)) revert WrongCaller();
@@ -1127,13 +1127,9 @@ contract KlerosCore is IArbitratorV2 {
     /// @param _alreadyTransferred True if the tokens were already transferred from juror. Only relevant for delayed stakes.
     /// @return succeeded True if the call succeeded, false otherwise.
     function _setStakeForAccount(
-        
         address _account,
-       
         uint96 _courtID,
-       
-        uint256 _newStake
-    ,
+        uint256 _newStake,
         bool _alreadyTransferred
     ) internal returns (bool succeeded) {
         if (_courtID == FORKING_COURT || _courtID > courts.length) return false;
@@ -1159,7 +1155,7 @@ contract KlerosCore is IArbitratorV2 {
         if (_newStake >= currentStake) {
             if (!_alreadyTransferred) {
                 // Stake increase
-            // When stakedPnk becomes lower than lockedPnk count the locked tokens in when transferring tokens from juror.
+                // When stakedPnk becomes lower than lockedPnk count the locked tokens in when transferring tokens from juror.
                 // (E.g. stakedPnk = 0, lockedPnk = 150) which can happen if the juror unstaked fully while having some tokens locked.
                 uint256 previouslyLocked = (juror.lockedPnk >= juror.stakedPnk) ? juror.lockedPnk - juror.stakedPnk : 0; // underflow guard
                 transferredAmount = (_newStake >= currentStake + previouslyLocked) // underflow guard
@@ -1202,13 +1198,15 @@ contract KlerosCore is IArbitratorV2 {
         // Note that stakedPnk can become async with currentStake (e.g. after penalty).
         // Also note that these values were already updated if the stake was only partially delayed.
         if (!_alreadyTransferred) {
-            juror.stakedPnk = (juror.stakedPnk >= currentStake) ? juror.stakedPnk - currentStake + _newStake : _newStake;
+            juror.stakedPnk = (juror.stakedPnk >= currentStake)
+                ? juror.stakedPnk - currentStake + _newStake
+                : _newStake;
             juror.stakedPnkByCourt[_courtID] = _newStake;
         }
 
         // Transfer the tokens but don't update sortition module.
         if (result == ISortitionModule.preStakeHookResult.partiallyDelayed) {
-            emit StakePartiallyDelayed(_account, _courtID, _stake);
+            emit StakePartiallyDelayed(_account, _courtID, _newStake);
             return true;
         }