diff --git a/sdk/python/setup.sh b/sdk/python/setup.sh
index 71b17727f5..09b1a6c730 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,9 +14,7 @@ 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://docsupport.blob.core.windows.net/ml-sample-submissions/3948399/azure_ai_ml-1.18.0-py3-none-any.whl
 # </az_ml_sdk_test_install>
 
 pip list
\ No newline at end of file