minor tweak to code styling