diff --git a/samples/docker/Dockerfile b/samples/docker/Dockerfile
index 90beab862..3f4f8e9c2 100644
--- a/samples/docker/Dockerfile
+++ b/samples/docker/Dockerfile
@@ -1,9 +1,9 @@
 FROM python:3.7.0-slim
-ARG APP_VERSION
-COPY google_pso_data_validator-$APP_VERSION-py3-none-any.whl .
+ARG _APP_VERSION
+COPY google_pso_data_validator-$_APP_VERSION-py3-none-any.whl .
 RUN apt-get update \
 && apt-get install gcc -y \
 && apt-get clean
 RUN pip install --upgrade pip
-RUN pip install google_pso_data_validator-$APP_VERSION-py3-none-any.whl
+RUN pip install google_pso_data_validator-$_APP_VERSION-py3-none-any.whl
 ENTRYPOINT ["python","-m","data_validation"]
diff --git a/samples/docker/build_docker.sh b/samples/docker/build_docker.sh
index 812e69b4d..d85ae41e0 100755
--- a/samples/docker/build_docker.sh
+++ b/samples/docker/build_docker.sh
@@ -26,5 +26,5 @@ python setup.py register sdist bdist_wheel
 
 cd samples/docker
 cp ../../dist/google_pso_data_validator-${PACKAGE_VERSION}-py3-none-any.whl .
-docker build -t data_validation:$PACKAGE_VERSION --build-arg APP_VERSION=$PACKAGE_VERSION .
+docker build -t data_validation:$PACKAGE_VERSION --build-arg _APP_VERSION=$PACKAGE_VERSION .
 rm google_pso_data_validator-${PACKAGE_VERSION}-py3-none-any.whl
\ No newline at end of file