Fix removeClass call