Remove test code