Remove legacy test code