I've add an extra if condition 

I've added this to make sure they would have a bloaty head if their head wasn't changed, in case they die after being treated.