diff --git a/main/signals.py b/main/signals.py index 475b6e7..45ee911 100644 --- a/main/signals.py +++ b/main/signals.py @@ -1,10 +1,10 @@ -from django.contrib.auth.models import User +from django.contrib.auth import get_user_model from django.db.models.signals import post_save from django.dispatch import receiver from main.models import UserProfile -@receiver(post_save, sender=User, dispatch_uid='randomz') +@receiver(post_save, sender=get_user_model(), dispatch_uid='randomz') def create_user_profile(sender, instance, created, **kwargs): # Create a new UserProfile object when we create a new User. if created: