dust rounding

uniswapfoundation · Apr 9, 2024 · 09ab194 · 09ab194
1 parent 2ac42a5
commit 09ab194
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/UniStaker.sol b/src/UniStaker.sol
@@ -802,7 +802,8 @@ contract UniStaker is INotifiableRewardReceiver, Multicall, EIP712, Nonces {
 
     uint256 _reward = scaledUnclaimedRewardCheckpoint[_beneficiary] / SCALE_FACTOR;
     if (_reward == 0) return;
-    scaledUnclaimedRewardCheckpoint[_beneficiary] = 0;
+    scaledUnclaimedRewardCheckpoint[_beneficiary] =
+      scaledUnclaimedRewardCheckpoint[_beneficiary] - (_reward * SCALE_FACTOR);
     emit RewardClaimed(_beneficiary, _reward);
 
     SafeERC20.safeTransfer(REWARD_TOKEN, _beneficiary, _reward);