diff --git a/test_coffeescript.py b/test_coffeescript.py
index <HASH>..<HASH> 100755
--- a/test_coffeescript.py
+++ b/test_coffeescript.py
@@ -1,14 +1,20 @@
 #!/usr/bin/env python3
 # -*- encoding: utf-8 -*-
 from __future__ import unicode_literals, print_function
-import doctest
-import unittest
-
+import sys
 import os
 import io
 import tempfile
 from itertools import product
 
+if sys.version_info < (2, 7):
+    import unittest2 as unittest
+else:
+    import unittest
+
+import doctest
+
+
 import execjs
 import coffeescript
Added version check for unitest to test