diff --git a/how-to-use-azureml/training/train-on-amlcompute/train-on-amlcompute.ipynb b/how-to-use-azureml/training/train-on-amlcompute/train-on-amlcompute.ipynb
index 70d2d5713..c83b53285 100644
--- a/how-to-use-azureml/training/train-on-amlcompute/train-on-amlcompute.ipynb
+++ b/how-to-use-azureml/training/train-on-amlcompute/train-on-amlcompute.ipynb
@@ -338,7 +338,8 @@
         "                                                           admin_user_ssh_key='<my-sshkey>',\n",
         "                                                           remote_login_port_public_access='enabled',\n",
         "                                                           identity_type='UserAssigned',\n",
-        "                                                           identity_id=['<resource-id1>'])\n",
+        "                                                           identity_id=['<resource-id1>'],\n",
+        "                                                           location='<my-cluster-location'>)\n",
         "    cpu_cluster = ComputeTarget.create(ws, cpu_cluster_name, compute_config)\n",
         "\n",
         "cpu_cluster.wait_for_completion(show_output=True)"
@@ -493,4 +494,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 2
-}
\ No newline at end of file
+}