diff --git a/hyperparameter_tuning/keras_bring_your_own/main.py b/hyperparameter_tuning/keras_bring_your_own/main.py
index 8234c0e2bd..16e8a0beb7 100644
--- a/hyperparameter_tuning/keras_bring_your_own/main.py
+++ b/hyperparameter_tuning/keras_bring_your_own/main.py
@@ -80,7 +80,7 @@ def upload_training_data():
 
     tensorflow_version_tag = get_tensorflow_version_tag(args.tf_version, args.instance_type)
 
-    image_name = get_image_name(args.ecr_repository, args.tensorflow_version_tag)
+    image_name = get_image_name(args.ecr_repository, tensorflow_version_tag)
 
     build_image(image_name, tensorflow_version_tag)