diff --git a/mfa/U2F.py b/mfa/U2F.py
index fe5ea19..0eb04f0 100644
--- a/mfa/U2F.py
+++ b/mfa/U2F.py
@@ -12,6 +12,7 @@ from django.conf import settings
 from django.http import HttpResponse
 from .models import *
 from .views import login
+from .Common import get_redirect_url
 import datetime
 from django.utils import timezone
 
diff --git a/mfa/totp.py b/mfa/totp.py
index 96617c7..76aa810 100644
--- a/mfa/totp.py
+++ b/mfa/totp.py
@@ -1,6 +1,7 @@
 from django.shortcuts import render
 from django.views.decorators.cache import never_cache
 from django.http import HttpResponse
+from .Common import get_redirect_url
 from .models import *
 from django.template.context_processors import csrf
 import simplejson