diff --git a/custom-user-model-ex1/user/models.py b/custom-user-model-ex1/user/models.py
index fc9bd77..efecc4d 100644
--- a/custom-user-model-ex1/user/models.py
+++ b/custom-user-model-ex1/user/models.py
@@ -15,9 +15,10 @@ class Role(models.TextChoices):
     role = models.CharField(max_length=50, choices=Role.choices)
 
     def save(self, *args, **kwargs):
-        if not self.pk:
+        if not self.role:
             self.role = self.base_role
             return super().save(*args, **kwargs)
+        return super().save(*args, **kwargs)
 
 
 class StudentManager(BaseUserManager):