Modify test code.