Add in a try catch and remove the die