fix ValueError in tests

jazzband · Mar 13, 2020 · fa2a487 · fa2a487
1 parent b1f9683
commit fa2a487
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/categories/fields.py b/categories/fields.py
@@ -3,18 +3,16 @@
 
 class CategoryM2MField(ManyToManyField):
     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):
     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)
 
 
 try: