diff --git a/backend/time_tracking/models.py b/backend/time_tracking/models.py
index d85ed99..514d1e2 100644
--- a/backend/time_tracking/models.py
+++ b/backend/time_tracking/models.py
@@ -1,9 +1,9 @@
 from django.db import models
-from django.conf import settings
-from datetime import datetime, date
+from django.contrib.auth import get_user_model
+from datetime import datetime
 
 class TimeEntry(models.Model):
-    user = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE, related_name='time_entries')
+    user = models.ForeignKey(get_user_model(), on_delete=models.CASCADE, related_name='time_entries')
     date = models.DateField()
     start_time = models.TimeField(null=True, blank=True)
     end_time = models.TimeField(null=True, blank=True)