diff --git a/categories/fields.py b/categories/fields.py index 1f44d1f..25f77d2 100644 --- a/categories/fields.py +++ b/categories/fields.py @@ -7,19 +7,15 @@ class CategoryM2MField(ManyToManyField): """A many to many field to a Category model.""" def __init__(self, **kwargs): - from .models import Category - if "to" in kwargs: kwargs.pop("to") - super(CategoryM2MField, self).__init__(to=Category, **kwargs) + super(CategoryM2MField, self).__init__(to="categories.Category", **kwargs) class CategoryFKField(ForeignKey): """A foreign key to the Category model.""" def __init__(self, **kwargs): - from .models import Category - if "to" in kwargs: kwargs.pop("to") - super(CategoryFKField, self).__init__(to=Category, **kwargs) + super(CategoryFKField, self).__init__(to="categories.Category", **kwargs)