feat: apply PR reviews

kroma-network · Jul 22, 2024 · 533bd67 · 533bd67
1 parent 23631c9
commit 533bd67
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/packages/contracts/contracts/L1/AssetManager.sol b/packages/contracts/contracts/L1/AssetManager.sol
@@ -744,7 +744,11 @@ contract AssetManager is ISemver, IERC721Receiver, IAssetManager {
         uint128 validatorReward
     ) external onlyValidatorManager {
         // Distribute the reward from a designated vault to the AssetManager contract.
-        ASSET_TOKEN.transferFrom(VALIDATOR_REWARD_VAULT, address(this), baseReward + boostedReward + validatorReward);
+        ASSET_TOKEN.safeTransferFrom(
+            VALIDATOR_REWARD_VAULT,
+            address(this),
+            baseReward + boostedReward + validatorReward
+        );
 
         // If reward is distributed to SECURITY_COUNCIL, transfer it directly.
         if (validator == SECURITY_COUNCIL) {
@@ -758,6 +762,7 @@ contract AssetManager is ISemver, IERC721Receiver, IAssetManager {
                 asset.totalKro += baseReward;
                 asset.validatorKro += validatorReward;
                 asset.rewardPerKghStored += boostedReward / asset.totalKgh;
+                asset.validatorKroBonded -= BOND_AMOUNT;
             }
 
             emit ValidatorKroUnbonded(