diff --git a/python/CustomVision/ImageClassification/CustomVisionQuickstart.py b/python/CustomVision/ImageClassification/CustomVisionQuickstart.py
index 8bffaa66..439dd68b 100644
--- a/python/CustomVision/ImageClassification/CustomVisionQuickstart.py
+++ b/python/CustomVision/ImageClassification/CustomVisionQuickstart.py
@@ -29,7 +29,7 @@
 training_key = os.environ["VISION_TRAINING_KEY"]
 prediction_key = os.environ["VISION_PREDICTION_KEY"]
 prediction_resource_id = os.environ["VISION_PREDICTION_RESOURCE_ID"]
-
+Predict_ENDPOINT = os.environ["VISION_PREDICTION_ENDPOINT"]
 # </snippet_creds>
 
 # <snippet_auth>
@@ -101,7 +101,7 @@
 # <snippet_test>
 # Now there is a trained endpoint that can be used to make a prediction
 prediction_credentials = ApiKeyCredentials(in_headers={"Prediction-key": prediction_key})
-predictor = CustomVisionPredictionClient(ENDPOINT, prediction_credentials)
+predictor = CustomVisionPredictionClient(Predict_ENDPOINT, prediction_credentials)
 
 with open(os.path.join (base_image_location, "Test/test_image.jpg"), "rb") as image_contents:
     results = predictor.classify_image(