edit test code