Tiny code reformat