modify test code.