diff --git a/.github/workflows/publish_docker.yml b/.github/workflows/publish_docker.yml
index 8253b5f..3c3042b 100644
--- a/.github/workflows/publish_docker.yml
+++ b/.github/workflows/publish_docker.yml
@@ -1,6 +1,10 @@
 on:
   workflow_call:
     inputs:
+      checkout-code:
+        required: false
+        type: boolean
+        default: false
       artifact-skip:
         required: false
         type: boolean
@@ -51,6 +55,10 @@ jobs:
     env:
       RUNNER_TOOL_CACHE: /toolcache
     steps:
+      - name: Check out code
+        uses: actions/checkout@v3
+        if: ${{ inputs.checkout-code == true }}
+          
       - uses: actions/download-artifact@v3
         if: ${{ inputs.artifact-skip != true }}
         with: