modify test generator