diff --git a/.github/workflows/release.yaml b/.github/workflows/release.yaml
index ec5b702..b6c31ed 100644
--- a/.github/workflows/release.yaml
+++ b/.github/workflows/release.yaml
@@ -19,7 +19,7 @@ jobs:
         env:
           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
 
-      - uses: iterative/setup-cml@v1
+      - uses: iterative/setup-cml@v2.0.1
 
       - name: Create PR for Documentation
         id: push_image_info