Make sure we reset the state correctly when you die