Modify test code2