Remove unused test code