diff --git a/vmaig_auth/forms.py b/vmaig_auth/forms.py
index a3f6e92..fb9f250 100644
--- a/vmaig_auth/forms.py
+++ b/vmaig_auth/forms.py
@@ -2,7 +2,10 @@
 from django import forms
 from vmaig_auth.models import VmaigUser
 from django.contrib.auth.tokens import default_token_generator
-from django.contrib.sites.models import get_current_site
+try:
+    from django.contrib.sites.shortcuts import get_current_site
+except:
+    from django.contrib.sites.models import get_current_site
 from django.utils.http import urlsafe_base64_encode
 from django.utils.encoding import force_bytes
 from django.core.mail import send_mail
diff --git a/vmaig_auth/views.py b/vmaig_auth/views.py
index 0f205bb..64142ae 100644
--- a/vmaig_auth/views.py
+++ b/vmaig_auth/views.py
@@ -8,7 +8,10 @@
 from django.contrib import auth
 from django.contrib.auth.forms import PasswordChangeForm, SetPasswordForm
 from django.contrib.auth.tokens import default_token_generator
-from django.contrib.sites.models import get_current_site
+try:
+    from django.contrib.sites.shortcuts import get_current_site
+except:
+    from django.contrib.sites.models import get_current_site
 from django.utils.http import (base36_to_int, is_safe_url,
                                urlsafe_base64_decode, urlsafe_base64_encode)
 from vmaig_auth.forms import VmaigUserCreationForm, VmaigPasswordRestForm