@@ -295,8 +295,8 @@ contract ERC20FriendlyRewardModule is ERC20BaseRewardModule {
295295 bytes calldata data
296296 ) external override onlyOwner returns (uint256 spent , uint256 vested ) {
297297 _update ();
298- _unstake (account, user, shares);
299- (spent, vested ) = _stake (account, user, shares, data);
298+ (, vested) = _unstake (account, user, shares);
299+ (spent, ) = _stake (account, user, shares, data);
300300 }
301301
302302 /**
@@ -312,9 +312,8 @@ contract ERC20FriendlyRewardModule is ERC20BaseRewardModule {
312312 uint256 rewardTally
313313 ) internal view returns (uint256 ) {
314314 return
315- ((rewardsPerStakedShare - rewardTally) * shares * bonus) /
316- 10 ** DECIMALS / // counteract bonus coefficient
317- 10 ** DECIMALS; // counteract rewardsPerStakedShare coefficient
315+ ((((rewardsPerStakedShare - rewardTally) * shares) / 10 ** DECIMALS) * // counteract rewardsPerStakedShare coefficient
316+ bonus) / 10 ** DECIMALS; // counteract bonus coefficient
318317 }
319318
320319 /**
@@ -339,15 +338,17 @@ contract ERC20FriendlyRewardModule is ERC20BaseRewardModule {
339338 /**
340339 * @inheritdoc IRewardModule
341340 */
342- function update (address ) external override onlyOwner {
341+ function update (address ) external override {
342+ requireOwner ();
343343 _update ();
344344 }
345345
346346 /**
347347 * @notice method called ad hoc to clean up and perform additional accounting
348348 * @dev will only be called manually, and should not contain any essential logic
349349 */
350- function clean () external override onlyOwner {
350+ function clean () external override {
351+ requireOwner ();
351352 _update ();
352353 _clean (address (_token));
353354 }
0 commit comments