diff --git a/packages/vllm/Dockerfile b/packages/vllm/Dockerfile
index 8485171e7..9d84d45bb 100755
--- a/packages/vllm/Dockerfile
+++ b/packages/vllm/Dockerfile
@@ -1,7 +1,7 @@
 ARG LOCAL_VERSION
-FROM ghcr.io/defenseunicorns/leapfrogai/leapfrogai-sdk:${LOCAL_VERSION} as sdk
+FROM ghcr.io/defenseunicorns/leapfrogai/leapfrogai-sdk:${LOCAL_VERSION} AS sdk
 
-FROM nvidia/cuda:12.2.2-devel-ubuntu22.04 as builder
+FROM nvidia/cuda:12.2.2-devel-ubuntu22.04 AS builder
 ARG SDK_DEST=src/leapfrogai_sdk/build
 
 # Set the config file defaults
@@ -129,4 +129,4 @@ RUN python3 packages/vllm/src/model_download.py
 
 EXPOSE 50051:50051
 
-ENTRYPOINT ["python", "-m", "leapfrogai_sdk.cli", "--app-dir=packages/vllm/src/", "main:Model"]
\ No newline at end of file
+ENTRYPOINT ["python", "-m", "leapfrogai_sdk.cli", "--app-dir=packages/vllm/src/", "main:Model"]