diff --git a/.github/workflows/code_review.yml b/.github/workflows/code_review.yml
index d86ba05..e67e631 100644
--- a/.github/workflows/code_review.yml
+++ b/.github/workflows/code_review.yml
@@ -12,7 +12,7 @@ jobs:
       - name: Checkout repository
         uses: actions/checkout@v3
       - name: Code Review
-        uses: cds-snc/cds-ai-codereviewer@main
+        uses: cds-snc/cds-ai-codereviewer@feature/azure-open-ai
         with:
           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
           OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}