diff --git a/Dockerfile.podman b/Dockerfile.podman
index dcd21b3..843dc44 100644
--- a/Dockerfile.podman
+++ b/Dockerfile.podman
@@ -1,4 +1,6 @@
 ARG GOIMAGE
+ARG BASEIMAGE
+
 # Stage to build the driver
 FROM $GOIMAGE as builder
 RUN mkdir -p /go/src
@@ -14,7 +16,7 @@ RUN go run core/semver/semver.go -f mk
 
 # Dockerfile to build Unity CSI Driver
 # Fetching the base ubi micro image with the require packges committed using buildah
-FROM BASEIMAGE as driver
+FROM $BASEIMAGE as driver
 
 COPY --from=builder /go/src/csi-unity/bin/csi-unity /
 COPY csi-unity/scripts/run.sh /