Spending psi points based on the unmodified roll isn't very 5e friendly. Let's try a different approach:
"At 20th level, your ability to foresee the different possibilities has suffused you. When you roll an ability check, attack roll, or saving throw, you can choose to reroll the die and must...