test: Fix to indicate staking address on test

planetarium · Oct 18, 2024 · b7cbc03 · b7cbc03
1 parent a7325e0
commit b7cbc03
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.Lib9c.Tests/Action/ValidatorDelegation/ClaimRewardValidatorTest.cs b/.Lib9c.Tests/Action/ValidatorDelegation/ClaimRewardValidatorTest.cs
@@ -9,6 +9,7 @@ namespace Lib9c.Tests.Action.ValidatorDelegation;
 using Libplanet.Types.Assets;
 using Nekoyume.Action.ValidatorDelegation;
 using Nekoyume.Model.Guild;
+using Nekoyume.Model.Stake;
 using Nekoyume.Module.Guild;
 using Nekoyume.TypedAddress;
 using Nekoyume.ValidatorDelegation;
@@ -251,7 +252,7 @@ var expectedProposerReward
         var guildRepository = new GuildRepository(world, actionContext);
         var delegatee = validatorRepository.GetValidatorDelegatee(validatorKey.Address);
         var actualRemainReward = world.GetBalance(delegatee.RewardRemainderPoolAddress, RewardCurrency);
-        var actualValidatorBalance = world.GetBalance(validatorKey.Address, DelegationCurrency);
+        var actualValidatorBalance = world.GetBalance(StakeState.DeriveAddress(validatorKey.Address), DelegationCurrency);
         var actualValidatorReward = world.GetBalance(validatorKey.Address, RewardCurrency);
         var actualDelegatorBalances = delegatorKeys
             .Select(item => world.GetBalance(item.Address, DelegationCurrency))