you've got it right — if a player successfully steals a die and adds +1 to their count, just use the next one lost for that stolen die prompt. (that way the last prompt is still held for when they lose their final die.)