Minor cosmetic change to code