diff --git a/.github/workflows/ci_cd.yml b/.github/workflows/ci_cd.yml
index f055ded..ace10d9 100644
--- a/.github/workflows/ci_cd.yml
+++ b/.github/workflows/ci_cd.yml
@@ -82,7 +82,7 @@ jobs:
       - name: Checkout repository
         uses: actions/checkout@v3
         with:
-          ref: main
+          ref: ${{ github.head_ref }}
 
       - name: Set up Python ${{ matrix.python-version }}
         uses: actions/setup-python@v4
@@ -171,7 +171,7 @@ jobs:
       - name: Checkout repository
         uses: actions/checkout@v3
         with:
-          ref: main
+          ref: ${{ github.head_ref }}
 
       - name: Log in to the Container registry
         uses: docker/login-action@v2