diff --git a/.github/workflows/build_documentation.yml b/.github/workflows/build_documentation.yml
index 6598a65..0d82837 100644
--- a/.github/workflows/build_documentation.yml
+++ b/.github/workflows/build_documentation.yml
@@ -10,7 +10,8 @@ jobs:
     uses: huggingface/doc-builder/.github/workflows/build_main_documentation.yml@main
     with:
       commit_sha: ${{ github.sha }}
-      package: deep-rl-course
+      package: deep-rl-class
+      package_name: deep-rl-course
       path_to_docs: deep-rl-class/units/
       additional_args: --not_python_module
       languages: en
diff --git a/.github/workflows/build_pr_documentation.yml b/.github/workflows/build_pr_documentation.yml
index 3865b11..8ee457c 100644
--- a/.github/workflows/build_pr_documentation.yml
+++ b/.github/workflows/build_pr_documentation.yml
@@ -13,7 +13,8 @@ jobs:
     with:
       commit_sha: ${{ github.event.pull_request.head.sha }}
       pr_number: ${{ github.event.number }}
-      package: deep-rl-course
+      package: deep-rl-class
+      package_name: deep-rl-course
       path_to_docs: deep-rl-class/units/
       additional_args: --not_python_module
       languages: en