diff --git a/test/UniStaker.t.sol b/test/UniStaker.t.sol
index c56f993..474d16c 100644
--- a/test/UniStaker.t.sol
+++ b/test/UniStaker.t.sol
@@ -4903,7 +4903,7 @@ contract ClaimReward is UniStakerRewardsTest {
   function testFuzz_ReturnsClaimedRewardAmount(
     address _depositor,
     address _delegatee,
-    uint256 _stakeAmount,
+    uint96 _stakeAmount,
     uint256 _rewardAmount,
     uint256 _durationPercent
   ) public {
@@ -5033,7 +5033,7 @@ contract ClaimRewardOnBehalf is UniStakerRewardsTest {
   function testFuzz_ReturnsClaimedRewardAmount(
     uint256 _beneficiaryPrivateKey,
     address _sender,
-    uint256 _depositAmount,
+    uint96 _depositAmount,
     uint256 _durationPercent,
     uint256 _rewardAmount,
     address _delegatee,