Minor code reformat