fix test generator