diff --git a/.github/workflows/cml.yaml b/.github/workflows/cml.yaml
index deb93bec..fa8d2bc8 100644
--- a/.github/workflows/cml.yaml
+++ b/.github/workflows/cml.yaml
@@ -19,7 +19,7 @@ jobs:
           REPO_TOKEN: ${{ secrets.GITHUB_TOKEN }}
         run: |
           cml ci
-          if test ${{ github.ref_name }} = main; then
+          if test $GITHUB_REF_NAME == main; then
             PREVIOUS_REF=HEAD~1
           else
             PREVIOUS_REF=main