diff --git a/sdk/python/setup.sh b/sdk/python/setup.sh
index ec29a18244..5b5a73321a 100644
--- a/sdk/python/setup.sh
+++ b/sdk/python/setup.sh
@@ -1,7 +1,6 @@
 #!/bin/bash
 
 # <az_ml_sdk_install>
-# pip install --pre azure-ai-ml
 # </az_ml_sdk_install>
 
 # <mldesigner_install>
@@ -15,12 +14,11 @@ pip install pandas
 
 
 # <az_ml_sdk_test_install>
-# pip install azure-ai-ml==0.1.0.b8
-pip install azure-ai-ml
-# https://docsupport.blob.core.windows.net/ml-sample-submissions/1905732/azure_ai_ml-1.0.0-py3-none-any.whl
+pip install https://azuresdkartifacts.z5.web.core.windows.net/python/distributions/ml-sample/6528728/azure_ai_ml-1.35.0-py3-none-any.whl || pip install azure-ai-ml
 # </az_ml_sdk_test_install>
 
 # protobuf==5.29.0 has IndentationError bug
 pip install "protobuf<=5.28.3"
 
-pip list
\ No newline at end of file
+
+pip list