Remove some test code