Happy to see that you enjoyed my game! I think the pity system could be a good idea. One of the previous comments said something about having a minimum dice upgrade. This may also solve the trouble you were having.
