diff --git a/apps/collection/models.py b/apps/collection/models.py index 71fb2052..f57e7c86 100644 --- a/apps/collection/models.py +++ b/apps/collection/models.py @@ -28,9 +28,9 @@ class CollectionDateMixin(models.Model): class CollectionQuerySet(models.QuerySet): """QuerySet for model Collection""" - def by_country(self, country_id): - """Filter collection by country id.""" - return self.filter(country=country_id) + def by_country(self, country): + """Filter collection by country.""" + return self.filter(country=country) def published(self): """Returned only published collection"""