diff --git a/tests/test_dereference.py b/tests/test_dereference.py index 64ddf09b..120b1a26 100644 --- a/tests/test_dereference.py +++ b/tests/test_dereference.py @@ -1,6 +1,8 @@ from __future__ import with_statement import unittest +from bson import DBRef + from mongoengine import * from mongoengine.connection import get_db from mongoengine.tests import query_counter