update the test code