Set klass onto widget, don't return it.