modify test code