Fix test which was failing do to CSS class name change.