diff --git a/use_live_db/test_runner.py b/use_live_db/test_runner.py
index b42dea1..2c0b198 100644
--- a/use_live_db/test_runner.py
+++ b/use_live_db/test_runner.py
@@ -1,9 +1,9 @@
 # Based on http://stackoverflow.com/a/5809952/689985
 
-from django.test.simple import DjangoTestSuiteRunner
-from django.test.simple import dependency_ordered
+from django.test.runner import DiscoverRunner
+from django.test.runner import dependency_ordered
 
-class ByPassableDBDjangoTestSuiteRunner(DjangoTestSuiteRunner):
+class ByPassableDBDjangoTestSuiteRunner(DiscoverRunner):
     def setup_databases(self, **kwargs):
         from django.db import connections, DEFAULT_DB_ALIAS