Fixed test for new css classes