Update the test code.