diff --git a/admin_auto_tests/test_model.py b/admin_auto_tests/test_model.py index 607838c..5d41e27 100644 --- a/admin_auto_tests/test_model.py +++ b/admin_auto_tests/test_model.py @@ -2,7 +2,10 @@ import factory from django.contrib.auth import get_user_model -from model_mommy import mommy +try: + from model_bakery import bakery as mommy +except ImportError: + from model_mommy import mommy try: from django.core.urlresolvers import reverse