Change test code