diff --git a/setup.py b/setup.py
index 500bed6d2..9c6278e75 100644
--- a/setup.py
+++ b/setup.py
@@ -110,12 +110,10 @@ setup(
         'cors': ['flask-cors>=2.0.0'],
         'console_log': ['console_log==0.2.10'],
         'hive': [
-            'pyhive>=0.6.1',
+            'pyhive[hive]>=0.6.1',
             'tableschema',
-            'thrift-sasl>=0.2.1',
-            'thrift>=0.9.3',
         ],
-        'presto': ['pyhive>=0.4.0'],
+        'presto': ['pyhive[presto]>=0.4.0'],
         'gsheets': ['gsheetsdb>=0.1.9'],
     },
     author='Apache Software Foundation',