adapt to make_generator_source change