Small code cleanup (reformat)