diff --git a/setup.py b/setup.py
index 3f361f82c3..a9155c8189 100644
--- a/setup.py
+++ b/setup.py
@@ -367,6 +367,7 @@ def run(self):
         "docs": DOCS_REQUIRED,
         "cassandra": CASSANDRA_REQUIRED,
         "hazelcast": HAZELCAST_REQUIRED,
+        "rockset": ROCKSET_REQUIRED,
     },
     include_package_data=True,
     license="Apache",