diff --git a/cv/aml-cli-v2/mlops/github-actions/deploy-online-endpoint-pipeline.yml b/cv/aml-cli-v2/mlops/github-actions/deploy-online-endpoint-pipeline.yml
index 156adab..0f3b8b6 100644
--- a/cv/aml-cli-v2/mlops/github-actions/deploy-online-endpoint-pipeline.yml
+++ b/cv/aml-cli-v2/mlops/github-actions/deploy-online-endpoint-pipeline.yml
@@ -27,7 +27,7 @@ jobs:
       file_name: ${{ needs.set-env-branch.outputs.config-file}}
   create-endpoint:
     needs: get-config
-    uses: Azure/mlops-templates/.github/workflows/cli-v2-create-endpoint.yml@main-dec31 # TODO - revert to @main
+    uses: Azure/mlops-templates/.github/workflows/create-endpoint.yml@main-dec31 # TODO - revert to @main
     with:
       resource_group: ${{ needs.get-config.outputs.resource_group }}
       workspace_name: ${{ needs.get-config.outputs.aml_workspace }}
@@ -37,7 +37,7 @@ jobs:
     secrets:
       creds: ${{secrets.AZURE_CREDENTIALS}}
   create-deployment:
-    uses: Azure/mlops-templates/.github/workflows/cli-v2-create-deployment.yml@main-dec31 # TODO - revert to @main
+    uses: Azure/mlops-templates/.github/workflows/create-deployment.yml@main-dec31 # TODO - revert to @main
     needs: [get-config, create-endpoint]
     with:
       resource_group: ${{ needs.get-config.outputs.resource_group }}
@@ -49,7 +49,7 @@ jobs:
     secrets:
       creds: ${{secrets.AZURE_CREDENTIALS}}
   allocate-traffic:
-    uses: Azure/mlops-templates/.github/workflows/cli-v2-allocate-traffic.yml@main-dec31 # TODO - revert to @main
+    uses: Azure/mlops-templates/.github/workflows/allocate-traffic.yml@main-dec31 # TODO - revert to @main
     needs: [get-config, create-deployment]
     with:
       resource_group: ${{ needs.get-config.outputs.resource_group }}