diff --git a/setup.py b/setup.py
index 798318c7d99aa..4a63010b2da26 100644
--- a/setup.py
+++ b/setup.py
@@ -218,6 +218,10 @@ def write_version(filename=os.path.join(*['airflow',
     'rednose',
     'requests_mock'
 ]
+
+if not PY3:
+    devel += ['unittest2']
+
 devel_minreq = devel + kubernetes + mysql + doc + password + s3 + cgroups
 devel_hadoop = devel_minreq + hive + hdfs + webhdfs + kerberos
 devel_all = (sendgrid + devel + all_dbs + doc + samba + s3 + slack + crypto + oracle +
@@ -231,7 +235,7 @@ def write_version(filename=os.path.join(*['airflow',
     devel_ci = [package for package in devel_all if package not in
                 ['snakebite>=2.7.8', 'snakebite[kerberos]>=2.7.8']]
 else:
-    devel_ci = devel_all + ['unittest2']
+    devel_ci = devel_all
 
 
 def do_setup():