fixed tests after class and selector change