diff --git a/ml_service/util/create_scoring_image.py b/ml_service/util/create_scoring_image.py
index c985e933..dadaf12b 100644
--- a/ml_service/util/create_scoring_image.py
+++ b/ml_service/util/create_scoring_image.py
@@ -26,7 +26,7 @@
 args = parser.parse_args()
 
 model = Model(ws, name=e.model_name, version=e.model_version)
-sources_dir = e.sources_dir
+sources_dir = e.sources_directory_train
 if (sources_dir is None):
     sources_dir = 'code'
 path_to_scoring = os.path.join(".", sources_dir, "scoring")