diff --git a/problem_builder/tests/integration/base_test.py b/problem_builder/tests/integration/base_test.py
index 49156720..ca504ffc 100644
--- a/problem_builder/tests/integration/base_test.py
+++ b/problem_builder/tests/integration/base_test.py
@@ -17,7 +17,6 @@
 # along with this program in a file in the toplevel directory called
 # "AGPLv3".  If not, see <http://www.gnu.org/licenses/>.
 #
-import time
 
 from unittest import mock
 from xblock.fields import String
diff --git a/problem_builder/tests/integration/test_completion.py b/problem_builder/tests/integration/test_completion.py
index 71e1e062..6fb3922d 100644
--- a/problem_builder/tests/integration/test_completion.py
+++ b/problem_builder/tests/integration/test_completion.py
@@ -19,6 +19,7 @@
 #
 
 # Imports ###########################################################
+import time
 
 from .base_test import (GetChoices, MentoringAssessmentBaseTest,
                         ProblemBuilderBaseTest)