diff --git a/src/UniStaker.sol b/src/UniStaker.sol
index 019ee70..689b335 100644
--- a/src/UniStaker.sol
+++ b/src/UniStaker.sol
@@ -241,7 +241,7 @@ contract UniStaker is INotifiableRewardReceiver, Multicall, EIP712, Nonces {
   /// Note that the contract tracks the unclaimed rewards internally with the scale factor
   /// included, in order to avoid the accrual of precision losses as users takes actions that
   /// cause rewards to be checkpointed. This external helper method is useful for integrations, and
-  /// returns the value after it  has been scaled down to the reward token's raw decimal amount.
+  /// returns the value after it has been scaled down to the reward token's raw decimal amount.
   /// @return Live value of the unclaimed rewards earned by a given beneficiary account.
   function unclaimedReward(address _beneficiary) external view returns (uint256) {
     return _scaledUnclaimedReward(_beneficiary) / SCALE_FACTOR;
diff --git a/test/UniStaker.integration.t.sol b/test/UniStaker.integration.t.sol
index 4c0cee8..73f7b6f 100644
--- a/test/UniStaker.integration.t.sol
+++ b/test/UniStaker.integration.t.sol
@@ -265,7 +265,7 @@ contract Stake is IntegrationTest, PercentAssertions {
   ) public {
     vm.assume(_depositor != address(0) && _delegatee != address(0) && _amount != 0);
     // Make sure depositor is not UniStaker
-    vm.assume(_depositor != 0xE2307e3710d108ceC7a4722a020a050681c835b3);
+    vm.assume(_depositor != address(uniStaker));
     _passQueueAndExecuteProposals();
     _swapAndClaimFees(_swapAmount);
     _amount = _dealStakingToken(_depositor, _amount);