diff --git a/admin_auto_tests/test_model.py b/admin_auto_tests/test_model.py
index 607838c..0d59a58 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_baker import bakery as mommy
+except ImportError:
+    from model_mommy import mommy
 
 try:
     from django.core.urlresolvers import reverse