Slight code formatting tweak