removeClass fix