Remove the rest of the test code