Fix: special code was lost during update