diff --git a/action.yml b/action.yml
index f2f6428..9c99bec 100644
--- a/action.yml
+++ b/action.yml
@@ -27,13 +27,14 @@ runs:
     
     - name: Run Code Review GPT
       shell: bash
-      run: npx code-review-gpt review --ci=github --model=$MODEL
+      run: npx code-review-gpt review --ci=github --model=$MODEL --provider=$PROVIDER
       
       env:
         MODEL: ${{ inputs.MODEL }}
         OPENAI_API_KEY: ${{ inputs.OPENAI_API_KEY }}
         BASE_SHA: ${{ github.event.pull_request.base.sha }}
         GITHUB_TOKEN: ${{ inputs.GITHUB_TOKEN }}
+        PROVIDER: ${{ inputs.PROVIDER }}
 
 branding:
   icon: 'code'
diff --git a/packages/code-review-gpt/src/common/model/AIModel.ts b/packages/code-review-gpt/src/common/model/AIModel.ts
index 301e862..27ca75e 100644
--- a/packages/code-review-gpt/src/common/model/AIModel.ts
+++ b/packages/code-review-gpt/src/common/model/AIModel.ts
@@ -37,7 +37,6 @@ class AIModel {
           configuration: { organization: options.organization },
           azureOpenAIApiVersion: '2024-04-01-preview',
           azureOpenAIApiKey: options.apiKey,
-          // azureOpenAIApiInstanceName
           azureOpenAIApiDeploymentName: 'D-OAI-model-deploy',
           azureOpenAIBasePath: 'https://d-oai-dev.openai.azure.com',
         });