Some fixes when die