diff --git a/djadmin2/tests/test_core.py b/djadmin2/tests/test_core.py
index 8713543..15d0881 100644
--- a/djadmin2/tests/test_core.py
+++ b/djadmin2/tests/test_core.py
@@ -1,7 +1,6 @@
-import unittest
-
 from django.db import models
 from django.core.exceptions import ImproperlyConfigured
+from django.test import TestCase
 
 from ..models import ModelAdmin2
 from ..core import Admin2
@@ -11,7 +10,7 @@ class Thing(models.Model):
     pass
 
 
-class Admin2Test(unittest.TestCase):
+class Admin2Test(TestCase):
     def setUp(self):
         self.admin2 = Admin2()
 
diff --git a/example/blog/tests/test_apiviews.py b/example/blog/tests/test_apiviews.py
index 29c4f11..621b88a 100644
--- a/example/blog/tests/test_apiviews.py
+++ b/example/blog/tests/test_apiviews.py
@@ -1,4 +1,4 @@
-from django.utils import unittest
+from django.test import TestCase
 from django.test.client import RequestFactory
 from django.core.urlresolvers import reverse
 
@@ -9,7 +9,7 @@ from djadmin2.models import ModelAdmin2
 from ..models import Post
 
 
-class ViewTest(unittest.TestCase):
+class ViewTest(TestCase):
     def setUp(self):
         self.factory = RequestFactory()