diff --git a/wtforms_json/__init__.py b/wtforms_json/__init__.py index 38942b3..a4485c0 100644 --- a/wtforms_json/__init__.py +++ b/wtforms_json/__init__.py @@ -3,13 +3,20 @@ import six from wtforms import Form try: - from wtforms.ext.sqlalchemy.fields import ( + from wtforms_sqlalchemy.fields import ( QuerySelectField, QuerySelectMultipleField ) HAS_SQLALCHEMY_SUPPORT = True except ImportError: - HAS_SQLALCHEMY_SUPPORT = False + try: + from wtforms.ext.sqlalchemy.fields import ( + QuerySelectField, + QuerySelectMultipleField + ) + HAS_SQLALCHEMY_SUPPORT = True + except ImportError: + HAS_SQLALCHEMY_SUPPORT = False from wtforms.fields import ( _unset_value, BooleanField,