KCINFR-979: fixing generator test
Merge pull request #565 from leoherbie/fix_gen_test

KCINFR-979: fixing generator test