fix(generator): use correct return type to prevent compile errors (#158)

fix(generator): use correct return type to prevent compile errors (#158) #202