modify the test code