diff --git a/simple_sso/sso_client/client.py b/simple_sso/sso_client/client.py
index f9f40a0..48dc84a 100644
--- a/simple_sso/sso_client/client.py
+++ b/simple_sso/sso_client/client.py
@@ -4,14 +4,17 @@
 from django.urls import re_path
 from django.contrib.auth import login
 from django.contrib.auth.backends import ModelBackend
-from django.contrib.auth.models import User
 from django.http import HttpResponseRedirect
 from django.urls import NoReverseMatch, reverse
+from django.contrib.auth import get_user_model
 from django.views.generic import View
 from itsdangerous import URLSafeTimedSerializer
 from webservices.sync import SyncConsumer
 
 
+User = get_user_model()
+
+
 class LoginView(View):
     client = None