diff --git a/basic-template-dockerfile/{{cookiecutter.project_name}}/docker_build.sh b/basic-template-dockerfile/{{cookiecutter.project_name}}/docker_build.sh
index ec968ce..74a76b7 100644
--- a/basic-template-dockerfile/{{cookiecutter.project_name}}/docker_build.sh
+++ b/basic-template-dockerfile/{{cookiecutter.project_name}}/docker_build.sh
@@ -8,10 +8,10 @@ REGISTRY=""
 # SET the appname here
 PROJECT_NAME="{{ cookiecutter.project_name }}"
 
-while getopts a:r:v:h flag
+while getopts p:r:v:h flag
 do
     case "${flag}" in
-        a) PROJECT_NAME=${OPTARG};;
+        p) PROJECT_NAME=${OPTARG};;
         r) REGISTRY=${OPTARG};;
         v) VERSION=${OPTARG};;
         h) echo "Usage: ${0} [-h|[-p <project_name>][-r <registry_name>][-v <version>]]"
diff --git a/bayesian-optimization/{{cookiecutter.project_name}}/docker_build.sh b/bayesian-optimization/{{cookiecutter.project_name}}/docker_build.sh
index 83f9069..74a76b7 100755
--- a/bayesian-optimization/{{cookiecutter.project_name}}/docker_build.sh
+++ b/bayesian-optimization/{{cookiecutter.project_name}}/docker_build.sh
@@ -8,7 +8,7 @@ REGISTRY=""
 # SET the appname here
 PROJECT_NAME="{{ cookiecutter.project_name }}"
 
-while getopts a:r:v:h flag
+while getopts p:r:v:h flag
 do
     case "${flag}" in
         p) PROJECT_NAME=${OPTARG};;
diff --git a/mnist-training/{{cookiecutter.project_name}}/docker_build.sh b/mnist-training/{{cookiecutter.project_name}}/docker_build.sh
index 83f9069..74a76b7 100644
--- a/mnist-training/{{cookiecutter.project_name}}/docker_build.sh
+++ b/mnist-training/{{cookiecutter.project_name}}/docker_build.sh
@@ -8,7 +8,7 @@ REGISTRY=""
 # SET the appname here
 PROJECT_NAME="{{ cookiecutter.project_name }}"
 
-while getopts a:r:v:h flag
+while getopts p:r:v:h flag
 do
     case "${flag}" in
         p) PROJECT_NAME=${OPTARG};;