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

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