diff --git a/setup.py b/setup.py
index e220b3adb..da01f7fb9 100644
--- a/setup.py
+++ b/setup.py
@@ -43,7 +43,6 @@ install_requires = [
     "future>=0.16.0",
     "colorama>=0.3.9",
     "configobj>=5.0.6",
-    "networkx>=2.1",
     "gitpython>=2.1.8",
     "setuptools>=34.0.0",
     "nanotime>=0.5.2",
@@ -66,6 +65,11 @@ install_requires = [
     "win-unicode-console>=0.5; sys_platform == 'win32'",
 ]
 
+if sys.version_info[0] == 2:
+    install_requires.append("networkx>=2.1,<2.3")
+else:
+    install_requires.append("networkx>=2.1")
+
 # Extra dependencies for remote integrations
 gs = ["google-cloud-storage==1.13.0"]
 s3 = ["boto3==1.9.115"]