diff --git a/tests/storage/test_directory.py b/tests/storage/test_directory.py
index c4c321dda..9a80bf6c0 100644
--- a/tests/storage/test_directory.py
+++ b/tests/storage/test_directory.py
@@ -14,7 +14,7 @@
 # limitations under the License.
 
 
-from twisted.trial import unittest
+from tests import unittest
 from twisted.internet import defer
 
 from synapse.server import HomeServer
diff --git a/tests/storage/test_presence.py b/tests/storage/test_presence.py
index f3fab4fe7..9655d3cf4 100644
--- a/tests/storage/test_presence.py
+++ b/tests/storage/test_presence.py
@@ -14,7 +14,7 @@
 # limitations under the License.
 
 
-from twisted.trial import unittest
+from tests import unittest
 from twisted.internet import defer
 
 from synapse.server import HomeServer
diff --git a/tests/storage/test_profile.py b/tests/storage/test_profile.py
index 185527804..21df2babd 100644
--- a/tests/storage/test_profile.py
+++ b/tests/storage/test_profile.py
@@ -14,7 +14,7 @@
 # limitations under the License.
 
 
-from twisted.trial import unittest
+from tests import unittest
 from twisted.internet import defer
 
 from synapse.server import HomeServer
diff --git a/tests/storage/test_registration.py b/tests/storage/test_registration.py
index 5e11ce351..91e221d53 100644
--- a/tests/storage/test_registration.py
+++ b/tests/storage/test_registration.py
@@ -14,7 +14,7 @@
 # limitations under the License.
 
 
-from twisted.trial import unittest
+from tests import unittest
 from twisted.internet import defer
 
 from synapse.server import HomeServer
diff --git a/tests/storage/test_roommember.py b/tests/storage/test_roommember.py
index 5a6e93d4d..dd18291b4 100644
--- a/tests/storage/test_roommember.py
+++ b/tests/storage/test_roommember.py
@@ -14,7 +14,7 @@
 # limitations under the License.
 
 
-from twisted.trial import unittest
+from tests import unittest
 from twisted.internet import defer
 
 from synapse.server import HomeServer