Minor code layout fix