Remove unnecessary class from arrange tests.