Proper fix so it also works with with the widget backend.