fix metaclass inheritance

MongoEngine · Feb 8, 2016 · 92c9d0c · 92c9d0c
1 parent ba300aa
commit 92c9d0c
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/django_mongoengine/document.py b/django_mongoengine/document.py
@@ -34,20 +34,19 @@ class DjangoFlavor(object):
     _default_manager = QuerySetManager()
     serializable_value = serializable_value
     _get_pk_val = Model.__dict__["_get_pk_val"]
-    swap_base = True
 
     def __init__(self, *args, **kwargs):
         self._state = ModelState(self._meta.get("db_alias", me.DEFAULT_CONNECTION_NAME))
         super(DjangoFlavor, self).__init__(*args, **kwargs)
 
 class Document(django_meta(mtc.TopLevelDocumentMetaclass,
                            DjangoFlavor, me.Document)):
-    pass
+    swap_base = True
 
 class DynamicDocument(django_meta(mtc.TopLevelDocumentMetaclass,
                                   DjangoFlavor, me.DynamicDocument)):
-    pass
+    swap_base = True
 
 class EmbeddedDocument(django_meta(mtc.DocumentMetaclass,
                                    DjangoFlavor, me.EmbeddedDocument)):
-    pass
+    swap_base = True