Remove extra test code