edit test code.