diff --git a/InvenTree/build/test_api.py b/InvenTree/build/test_api.py
index 9551e3d07e..8c9d0fe7e1 100644
--- a/InvenTree/build/test_api.py
+++ b/InvenTree/build/test_api.py
@@ -5,6 +5,9 @@ from datetime import datetime, timedelta
 
 from django.urls import reverse
 
+from django.contrib.auth import get_user_model
+from django.contrib.auth.models import Group
+
 from rest_framework.test import APITestCase
 from rest_framework import status
 
diff --git a/InvenTree/build/tests.py b/InvenTree/build/tests.py
index dd1fe9b45f..e8ec8b67ca 100644
--- a/InvenTree/build/tests.py
+++ b/InvenTree/build/tests.py
@@ -3,6 +3,7 @@ from __future__ import unicode_literals
 
 from django.test import TestCase
 from django.urls import reverse
+
 from django.contrib.auth import get_user_model
 from django.contrib.auth.models import Group
 
diff --git a/InvenTree/order/test_views.py b/InvenTree/order/test_views.py
index 3af234e36a..f636d91fb9 100644
--- a/InvenTree/order/test_views.py
+++ b/InvenTree/order/test_views.py
@@ -8,12 +8,6 @@ from django.urls import reverse
 from django.contrib.auth import get_user_model
 from django.contrib.auth.models import Group
 
-from InvenTree.status_codes import PurchaseOrderStatus
-
-from .models import PurchaseOrder
-
-import json
-
 
 class OrderViewTestCase(TestCase):