Fix class names for image block