Remove unnecessary class