fix generator_test