minor comment fix in test_generator