Remove the test code