Decorate the queryset function