diff --git a/tests/fake_model.py b/tests/fake_model.py
index 128d77e..5f280e4 100644
--- a/tests/fake_model.py
+++ b/tests/fake_model.py
@@ -2,7 +2,8 @@ from django.db import connection, migrations
 from django.db.migrations.executor import MigrationExecutor
 from django.contrib.postgres.operations import HStoreExtension
 
-from localized_fields import LocalizedModel, AtomicSlugRetryMixin
+from localized_fields.models import LocalizedModel
+from localized_fields import AtomicSlugRetryMixin
 
 
 def define_fake_model(name='TestModel', fields=None):