Minor style tweak on generator