Fix die function