diff --git a/docker/Dockerfile.api b/docker/Dockerfile.api
index 0fe4d5308..31b85b459 100644
--- a/docker/Dockerfile.api
+++ b/docker/Dockerfile.api
@@ -1,6 +1,7 @@
 ARG base_image_tag
+ARG base_image
 
-FROM deepset/haystack:${base_image_tag}
+FROM ${base_image}:${base_image_tag}
 
 ENV SERVICE_NAME="gunicorn-service"
 
diff --git a/docker/docker-bake.hcl b/docker/docker-bake.hcl
index 7a63773f1..272094a3f 100644
--- a/docker/docker-bake.hcl
+++ b/docker/docker-bake.hcl
@@ -68,6 +68,7 @@ target "cpu" {
   dockerfile = "Dockerfile.api"
   tags = ["${IMAGE_NAME}:cpu-${IMAGE_TAG_SUFFIX}"]
   args = {
+    base_image = "${IMAGE_NAME}"
     base_image_tag = "base-cpu-${BASE_IMAGE_TAG_SUFFIX}"
   }
   platforms = ["linux/amd64", "linux/arm64"]
@@ -83,6 +84,7 @@ target "gpu" {
   dockerfile = "Dockerfile.api"
   tags = ["${IMAGE_NAME}:gpu-${IMAGE_TAG_SUFFIX}"]
   args = {
+    base_image = "${IMAGE_NAME}"
     base_image_tag = "base-gpu-${BASE_IMAGE_TAG_SUFFIX}"
   }
   platforms = ["linux/amd64", "linux/arm64"]