diff --git a/.github/workflows/docs-rebuild.yaml b/.github/workflows/docs-rebuild.yaml
index 826d726..21c9a27 100644
--- a/.github/workflows/docs-rebuild.yaml
+++ b/.github/workflows/docs-rebuild.yaml
@@ -4,6 +4,7 @@ on:
   push:
     branches:
       - main
+      - release-2.1
 
 jobs:
   docs-rebuild-deploy:
diff --git a/.github/workflows/test_agents.yaml b/.github/workflows/test_agents.yaml
index e3abcf9..0fb84e1 100644
--- a/.github/workflows/test_agents.yaml
+++ b/.github/workflows/test_agents.yaml
@@ -3,7 +3,7 @@ on:
 
 jobs:
   test:
-    uses: diambra/agents/.github/workflows/reusable-test.yaml@main
+    uses: diambra/agents/.github/workflows/reusable_unit_tests.yaml@main
     with:
       arena_requirement_specifier: 'git+https://github.com/diambra/arena.git@${{ github.ref }}#egg='
       agents_ref: 'main'