diff --git a/setup.py b/setup.py
index 3a2292759..518652dfe 100644
--- a/setup.py
+++ b/setup.py
@@ -49,6 +49,7 @@ class build_py(_build_py):
 
 
 install_requires = [
+    "networkx>=2.1,<2.3",
     "ply>=3.9",  # See https://github.com/pyinstaller/pyinstaller/issues/1945
     "configparser>=3.5.0",
     "future>=0.16.0",
@@ -78,11 +79,6 @@ 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.19.0"]