diff --git a/common/djangoapps/terrain/course_helpers.py b/common/djangoapps/terrain/course_helpers.py
index dbdaa2a21c..8c949de1ad 100644
--- a/common/djangoapps/terrain/course_helpers.py
+++ b/common/djangoapps/terrain/course_helpers.py
@@ -9,7 +9,6 @@ from django.contrib.sessions.middleware import SessionMiddleware
 from student.models import CourseEnrollment
 from bs4 import BeautifulSoup
 import os.path
-from selenium.common.exceptions import WebDriverException
 from urllib import quote_plus
 from lettuce.django import django_url
 
diff --git a/common/djangoapps/terrain/ui_helpers.py b/common/djangoapps/terrain/ui_helpers.py
index 4667957e87..2ad7150740 100644
--- a/common/djangoapps/terrain/ui_helpers.py
+++ b/common/djangoapps/terrain/ui_helpers.py
@@ -1,6 +1,7 @@
 from lettuce import world, step
 import time
 from urllib import quote_plus
+from selenium.common.exceptions import WebDriverException
 
 @world.absorb
 def wait(seconds):