Fix widget css_class