2 files changed, 10 insertions, 12 deletions
diff --git a/testing/test_soledad/u1db_tests/__init__.py b/testing/test_soledad/u1db_tests/__init__.py
index 36db548d..1575b859 100644
--- a/testing/test_soledad/u1db_tests/__init__.py
+++ b/testing/test_soledad/u1db_tests/__init__.py
@@ -28,12 +28,6 @@ import sys
 from six import StringIO
 from wsgiref import simple_server
 
-
-if sys.version_info[0] < 3:
-    from pysqlcipher import dbapi2
-else:
-    from pysqlcipher3 import dbapi2
-
 import testscenarios
 from twisted.trial import unittest
 from twisted.web.server import Site
@@ -48,6 +42,11 @@ from leap.soledad.common.l2db.remote import server_state
 from leap.soledad.common.l2db.remote import http_app
 from leap.soledad.common.l2db.remote import http_target
 
+if sys.version_info[0] < 3:
+    from pysqlcipher import dbapi2
+else:
+    from pysqlcipher3 import dbapi2
+
 
 class TestCase(unittest.TestCase):
 
diff --git a/testing/test_soledad/util.py b/testing/test_soledad/util.py
index 91291faf..6ffb60b6 100644
--- a/testing/test_soledad/util.py
+++ b/testing/test_soledad/util.py
@@ -31,11 +31,6 @@ from six import StringIO
 from uuid import uuid4
 from mock import Mock
 
-if sys.version_info[0] < 3:
-    from pysqlcipher import dbapi2
-else:
-    from pysqlcipher3 import dbapi2
-
 from twisted.trial import unittest
 
 from leap.common.testing.basetest import BaseLeapTest
@@ -48,7 +43,6 @@ from leap.soledad.common.document import SoledadDocument
 from leap.soledad.common.couch import CouchDatabase
 from leap.soledad.common.couch.state import CouchServerState
 
-
 from leap.soledad.client import Soledad
 from leap.soledad.client import http_target
 from leap.soledad.client import auth
@@ -58,6 +52,11 @@ from leap.soledad.client._crypto import is_symmetrically_encrypted
 
 from leap.soledad.server import SoledadApp
 
+if sys.version_info[0] < 3:
+    from pysqlcipher import dbapi2
+else:
+    from pysqlcipher3 import dbapi2
+
 
 PASSWORD = '123456'
 ADDRESS = 'user-1234'