clean up the test code