CSS classes fix