modify test code geneator