diff --git a/cli/setup.sh b/cli/setup.sh
index 8914f24309..e2ffa964cd 100644
--- a/cli/setup.sh
+++ b/cli/setup.sh
@@ -6,7 +6,7 @@
 ###  * Comment the ml extension install within <az_ml_install>
 
 # {
-#      wheel_url='https://azuremlsdktestpypi.blob.core.windows.net/wheels/sdk-cli-v2-public/ml-2.9.0-py3-none-any.whl'
+#      wheel_url='https://azuremlsdktestpypi.blob.core.windows.net/wheels/sdk-cli-v2-public/ml-2.40.1-py3-none-any.whl'
 #
 #      az extension remove -n ml
 #      if ! az extension add --yes --upgrade --source "$wheel_url"; then
@@ -44,4 +44,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