diff --git a/images/Dockerfile b/images/Dockerfile
index b3bffcee..4bafaa79 100644
--- a/images/Dockerfile
+++ b/images/Dockerfile
@@ -7,7 +7,6 @@ RUN apt-get update && \
     apt-get install git -y
 
 # Install Fondant
-# This is split from other requirements to leverage caching
 ARG FONDANT_VERSION=main
 RUN pip3 install fondant[component,aws,azure,gcp]@git+https://github.com/ml6team/fondant@${FONDANT_VERSION}
 
diff --git a/scripts/build_base_image.sh b/scripts/build_base_image.sh
index f832e9aa..1c41b95c 100755
--- a/scripts/build_base_image.sh
+++ b/scripts/build_base_image.sh
@@ -39,8 +39,8 @@ for python_version in "${python_versions[@]}"; do
 
     # Build docker images and push to docker hub
     docker build --push "${args[@]}" \
-    --build-arg PYTHON_VERSION=$python_version \
-    --build-arg FONDANT_VERSION=$tag \
+    --build-arg="PYTHON_VERSION=${python_version}" \
+    --build-arg="FONDANT_VERSION=${tag}" \
     -f "images/Dockerfile" \
    .
 done