diff --git a/onnxruntime/python/tools/transformers/notebooks/Inference_Bert_with_OnnxRuntime_on_AzureML.ipynb b/onnxruntime/python/tools/transformers/notebooks/Inference_Bert_with_OnnxRuntime_on_AzureML.ipynb
index bff136a2d9..bdbfbbbb6a 100644
--- a/onnxruntime/python/tools/transformers/notebooks/Inference_Bert_with_OnnxRuntime_on_AzureML.ipynb
+++ b/onnxruntime/python/tools/transformers/notebooks/Inference_Bert_with_OnnxRuntime_on_AzureML.ipynb
@@ -115,9 +115,9 @@
     "estimator = PyTorch(source_directory=project_roots, \n",
     "                    script_params={'--output-dir': './outputs'},\n",
     "                    compute_target=gpu_compute_target,\n",
-    "                    use_docker=True\n",
-    "                    custom_docker_image=image_name\n",
-    "                    script_params = {...}\n",
+    "                    use_docker=True,\n",
+    "                    custom_docker_image=image_name,\n",
+    "                    script_params = {...},\n",
     "                    entry_script='run_squad_azureml.py', # change here\n",
     "                    node_count=1,\n",
     "                    process_count_per_node=4,\n",