diff --git a/.github/workflows/publish-docs-to-s3.yml b/.github/workflows/publish-docs-to-s3.yml
index 15710d4e8a695..8c5d152931a94 100644
--- a/.github/workflows/publish-docs-to-s3.yml
+++ b/.github/workflows/publish-docs-to-s3.yml
@@ -138,7 +138,7 @@ jobs:
         run: |
           # Keep the non-provider list in sync with NON_PROVIDER_TOKENS in
           # dev/registry/derive_wave_providers.py.
-          NON_PROVIDER_TOKENS=("apache-airflow" "apache-airflow-ctl" "apache-airflow-task-sdk" "helm-chart" "docker-stack")
+          NON_PROVIDER_TOKENS=("apache-airflow" "apache-airflow-ctl" "task-sdk" "helm-chart" "docker-stack")
           has_provider=false
           for token in $INCLUDE_DOCS; do
             is_non_provider=false
diff --git a/dev/registry/derive_wave_providers.py b/dev/registry/derive_wave_providers.py
index 7a473e8a9ebe8..374cf3e6ff148 100644
--- a/dev/registry/derive_wave_providers.py
+++ b/dev/registry/derive_wave_providers.py
@@ -59,7 +59,7 @@
         "helm-chart",
         "docker-stack",
         "apache-airflow-ctl",
-        "apache-airflow-task-sdk",
+        "task-sdk",
     }
 )
 PROVIDER_PREFIX = "apache-airflow-providers-"