Get rid of test code