Remove unnecessary test code