diff --git a/setup.py b/setup.py
index 695933705b5..8133a2838a7 100644
--- a/setup.py
+++ b/setup.py
@@ -59,7 +59,8 @@ def package_files(directory):
         "cwrap",
         "numpy",
         "pandas",
-        "matplotlib<3",
+        "matplotlib < 3; python_version < '3.0'",
+        "matplotlib; python_version >= '3.0'",
         "scipy",
         "pytest",
         "decorator",