diff --git a/src/azure-cli-core/setup.py b/src/azure-cli-core/setup.py
index 0f89a7cf3c9..3b7772641fd 100644
--- a/src/azure-cli-core/setup.py
+++ b/src/azure-cli-core/setup.py
@@ -102,8 +102,6 @@
         ":python_version<'3.4'": ['enum34'],
         ":python_version<'2.7.9'": ['pyopenssl', 'ndg-httpsclient', 'pyasn1'],
         ':python_version<"3.0"': ['futures'],
-        ":python_version<'3.0'": ['antlr4-python2-runtime'],
-        ":python_version>='3.0'": ['antlr4-python3-runtime'],
         "test": TESTS_REQUIRE,
     },
     tests_require=TESTS_REQUIRE,
diff --git a/src/azure-cli/setup.py b/src/azure-cli/setup.py
index 7b3d3986bdf..8aab5c22f9f 100644
--- a/src/azure-cli/setup.py
+++ b/src/azure-cli/setup.py
@@ -50,6 +50,7 @@
 ]
 
 DEPENDENCIES = [
+    'antlr4-python3-runtime~=4.7.2',
     'azure-batch~=9.0',
     'azure-cli-command_modules-nspkg~=2.0',
     'azure-cli-core=={}.*'.format(VERSION),