clean up the test code
clean up the test code