Fixed small problem with the queryset