revise the test code