modify the test codes