diff --git a/docs/generate_api_reference.sh b/docs/generate_api_reference.sh
index 8a561028..fe1f0dad 100644
--- a/docs/generate_api_reference.sh
+++ b/docs/generate_api_reference.sh
@@ -2,7 +2,7 @@
 set -e  # Exit if any command fails
 
 # Install pdoc3 for generating Python API documentation
-pip install lavague gradio pdoc3
+pip install lavague pdoc3
 
 # Generate HTML documentation
-pdoc --html --skip-errors --template-dir docs/pdoc_template -o docs/docs/ src/lavague/ --force
+pdoc --html --skip-errors --template-dir docs/pdoc_template -o docs/docs/ lavague-core/lavague/core --force
diff --git a/mkdocs.yml b/mkdocs.yml
index 28c1f1a7..f6bc0b26 100644
--- a/mkdocs.yml
+++ b/mkdocs.yml
@@ -121,4 +121,4 @@ nav:
     - Documentation: "docs/contributing/documentation.md"
   
 - 🛠️ API reference:
-  - API reference home: 'docs/lavague/index.html'
\ No newline at end of file
+  - API reference home: 'docs/core/index.html'
\ No newline at end of file