Fix remove className bug