diff --git a/Dockerfile b/Dockerfile
index f48b832..020f14e 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,5 +1,5 @@
 ARG WORKER_CUDA_VERSION=11.8.0
-ARG BASE_IMAGE_VERSION=1.0.0
+ARG BASE_IMAGE_VERSION=1.0.0preview
 FROM runpod/worker-vllm:base-${BASE_IMAGE_VERSION}-cuda${WORKER_CUDA_VERSION} AS vllm-base
 
 RUN apt-get update -y \
diff --git a/docker-bake.hcl b/docker-bake.hcl
index f110a6d..fedfe81 100644
--- a/docker-bake.hcl
+++ b/docker-bake.hcl
@@ -43,7 +43,7 @@ target "base-1210" {
 }
 
 target "worker-1180" {
-  tags = ["${REPOSITORY}/worker-vllm:worker-${BASE_IMAGE_VERSION}-cuda11.8.0"]
+  tags = ["${REPOSITORY}/worker-vllm:${BASE_IMAGE_VERSION}-cuda11.8.0"]
   context = "."
   dockerfile = "Dockerfile"
   args = {
@@ -54,7 +54,7 @@ target "worker-1180" {
 }
 
 target "worker-1210" {
-  tags = ["${REPOSITORY}/worker-vllm:worker-${BASE_IMAGE_VERSION}-cuda12.1.0"]
+  tags = ["${REPOSITORY}/worker-vllm:${BASE_IMAGE_VERSION}-cuda12.1.0"]
   context = "."
   dockerfile = "Dockerfile"
   args = {