diff --git a/extras_mongoengine/__init__.py b/extras_mongoengine/__init__.py index 6a87ac0..49ec284 100644 --- a/extras_mongoengine/__init__.py +++ b/extras_mongoengine/__init__.py @@ -1,3 +1,3 @@ -import fields +import extras_mongoengine.fields as fields __all__ = ('fields') diff --git a/extras_mongoengine/fields.py b/extras_mongoengine/fields.py index 5871ddf..61f93fe 100644 --- a/extras_mongoengine/fields.py +++ b/extras_mongoengine/fields.py @@ -65,7 +65,7 @@ def validate(self, value): super(LowerEmailField, self).validate(value) -class EnumField(object): +class EnumField(BaseField): """ A class to register Enum type (from the package enum34) into mongo