diff --git a/bench/tests/test_base.py b/bench/tests/test_base.py
index 5807ec44..73cc90f5 100644
--- a/bench/tests/test_base.py
+++ b/bench/tests/test_base.py
@@ -1,19 +1,19 @@
 # imports - standard imports
+import getpass
 import json
 import os
 import shutil
 import subprocess
 import sys
+import traceback
 import unittest
-import getpass
-
-# imports - module imports
-import bench
-import bench.utils
 
 # imports - third party imports
 from six import PY2
 
+# imports - module imports
+import bench
+import bench.utils
 
 if PY2:
 	FRAPPE_BRANCH = "version-12"
diff --git a/bench/tests/test_init.py b/bench/tests/test_init.py
index 07b0f96a..69d49a45 100755
--- a/bench/tests/test_init.py
+++ b/bench/tests/test_init.py
@@ -11,7 +11,7 @@ import git
 import bench
 import bench.utils
 from bench.release import get_bumped_version
-from bench.tests.test_base import TestBenchBase, FRAPPE_BRANCH
+from bench.tests.test_base import FRAPPE_BRANCH, TestBenchBase
 
 
 class TestBenchInit(TestBenchBase):
@@ -155,4 +155,4 @@ class TestBenchInit(TestBenchBase):
 
 
 if __name__ == '__main__':
-	unittest.main()
\ No newline at end of file
+	unittest.main()
diff --git a/bench/tests/test_setup_production.py b/bench/tests/test_setup_production.py
index 9e9f0bd2..272db54d 100644
--- a/bench/tests/test_setup_production.py
+++ b/bench/tests/test_setup_production.py
@@ -176,4 +176,4 @@ class TestSetupProduction(TestBenchBase):
 
 
 if __name__ == '__main__':
-	unittest.main()
\ No newline at end of file
+	unittest.main()