diff --git a/djangoldp_notification/factories.py b/djangoldp_notification/factories.py
index 1910325b7263f047efa07c6f8dcbe0168de0b9ec..d3c985d703579301b442829ef5c7408f059dc638 100644
--- a/djangoldp_notification/factories.py
+++ b/djangoldp_notification/factories.py
@@ -1,6 +1,7 @@
 import factory
+from django.conf import settings
+
 from .models import Notification
-from django.contrib.auth.models import User
 from django.db.models.signals import post_save
 
 @factory.django.mute_signals(post_save)
@@ -11,7 +12,7 @@ class NotificationFactory(factory.django.DjangoModelFactory):
     type = factory.Faker('text', max_nb_chars=50)
     summary = factory.Faker('paragraph', nb_sentences=3, variable_nb_sentences=True)
     author_user = factory.Faker('url')
-    user = factory.Iterator(User.objects.all())
+    user = factory.Iterator(settings.AUTH_USER_MODEL.objects.all())
     date = factory.Faker('past_datetime')
     read = factory.Faker('boolean')
     object = factory.Faker('url')
\ No newline at end of file