diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml
index 4741f4117..c5ea5f6f7 100644
--- a/.github/workflows/docker-build.yml
+++ b/.github/workflows/docker-build.yml
@@ -85,6 +85,8 @@ jobs:
           context: .
           push: true
           platforms: "linux/amd64,linux/arm64/v8"
+          build-args: |
+            LANGFLOW_IMAGE=langflowai/langflow:${{ inputs.version }}
           file: ${{ matrix.dockerfile }}
           tags: ${{ matrix.tags }}