Merge pull request #151 from rma4ok/fix-form-params

[fix] referencing obj variable
MongoEngine · Jan 9, 2016 · 68ccb1e · 68ccb1e
2 parents 92d7c74 + 7a5d345
commit 68ccb1e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/flask_mongoengine/wtf/models.py b/flask_mongoengine/wtf/models.py
@@ -4,12 +4,12 @@
 class ModelForm(Form):
     """A WTForms mongoengine model form"""
 
-    def __init__(self, formdata=None, obj=None, prefix='', **kwargs):
-        self.instance = (kwargs.pop('instance', None) or kwargs.get('obj', None))
+    def __init__(self, formdata=None, obj=None, **kwargs):
+        self.instance = (kwargs.pop('instance', None) or obj)
         if self.instance and not formdata:
             obj = self.instance
         self.formdata = formdata
-        super(ModelForm, self).__init__(formdata, obj, prefix, **kwargs)
+        super(ModelForm, self).__init__(formdata, obj, **kwargs)
 
     def save(self, commit=True, **kwargs):
         if self.instance: