diff --git a/cli/setup.sh b/cli/setup.sh
index 8914f243093..a0d75476f4d 100644
--- a/cli/setup.sh
+++ b/cli/setup.sh
@@ -19,6 +19,20 @@
 #  }
 
 
+  {
+       wheel_url='https://azuremlsdktestpypi.blob.core.windows.net/wheels/sdk-cli-v2-public/ml-2.27.0-py3-none-any.whl'
+ 
+       az extension remove -n ml
+       if ! az extension add --yes --upgrade --source "$wheel_url"; then
+ 
+           echo "Error: Failed to install release candidate"
+           exit 1
+       fi
+       az version
+       unset wheel_url
+   }
+
+
 # <az_ml_install>
 az extension add -n ml -y
 # </az_ml_install>
@@ -44,4 +58,4 @@ then
     echo "Default resource group set to $GROUP"
 else
     echo "Workflows are using the new subscription."
-fi
\ No newline at end of file
+fi
diff --git a/sdk/python/setup.sh b/sdk/python/setup.sh
index 71b17727f51..5f8330ceb68 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/3860148/azure_ai_ml-1.17.0-py3-none-any.whl
 # </az_ml_sdk_test_install>
 
 pip list
\ No newline at end of file