diff --git a/.github/workflows/docker_build_push.yml b/.github/workflows/docker_build_push.yml
index b58bfa1..958b4ca 100644
--- a/.github/workflows/docker_build_push.yml
+++ b/.github/workflows/docker_build_push.yml
@@ -92,6 +92,7 @@ jobs:
       - name: Build and push Docker image
         uses: docker/build-push-action@v6.4.0
         with:
+          build-args: notebook_type=${{ inputs.NOTEBOOK_TYPE }}
           context: .
           push: true
           tags: ${{ matrix.value }}