Update sum-tree after prize awarded

pooltogether · Aug 11, 2020 · 2cada74 · 2cada74
1 parent 731eb69
commit 2cada74
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/contracts/prize-strategy/PrizeStrategy.sol b/contracts/prize-strategy/PrizeStrategy.sol
@@ -497,10 +497,12 @@ contract PrizeStrategy is PrizeStrategyStorage,
   /// @param user The user to whom the tickets are minted
   /// @param amount The amount of interest to mint as tickets.
   function _awardTickets(address user, uint256 amount) internal {
-    _accrueCredit(user, ticket.balanceOf(user));
+    uint256 userBalance = ticket.balanceOf(user);
+    _accrueCredit(user, userBalance);
     uint256 creditEarned = _calculateEarlyExitFee(amount);
     creditBalances[user].balance = uint256(creditBalances[user].balance).add(creditEarned).toUint128();
     prizePool.award(user, amount, address(ticket));
+    sortitionSumTrees.set(TREE_KEY, userBalance.add(amount), bytes32(uint256(user)));
   }
 
   /// @notice Awards all external tokens with non-zero balances to the given user.  The external tokens must be held by the PrizePool contract.