diff --git a/datahub-actions/setup.py b/datahub-actions/setup.py
index 6fffa70c..3d606fc9 100644
--- a/datahub-actions/setup.py
+++ b/datahub-actions/setup.py
@@ -33,7 +33,7 @@ def get_long_description():
 acryl_datahub_min_version = os.environ.get("ACRYL_DATAHUB_MIN_VERSION") or "0.12.1.2"
 
 base_requirements = {
-    f"acryl-datahub[kafka]>={acryl_datahub_min_version}",
+    f"acryl-datahub[datahub-kafka]>={acryl_datahub_min_version}",
     # Compatibility.
     "typing_extensions>=3.7.4; python_version < '3.8'",
     "mypy_extensions>=0.4.3",