diff --git a/tests/test_gallery.py b/tests/test_gallery.py
index 4c347f5..6d24025 100644
--- a/tests/test_gallery.py
+++ b/tests/test_gallery.py
@@ -2,7 +2,11 @@
 # -*- coding:utf-8 -*-
 
 import os
-import unittest
+
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
 
 from sigal.gallery import Gallery, get_metadata
 from sigal.settings import read_settings
diff --git a/tests/test_settings.py b/tests/test_settings.py
index ea80165..b26ab80 100644
--- a/tests/test_settings.py
+++ b/tests/test_settings.py
@@ -2,7 +2,11 @@
 # -*- coding:utf-8 -*-
 
 import os
-import unittest
+
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
 
 from sigal.settings import read_settings, get_size