fix the die function