diff --git a/cli/jobs/pipelines-with-components/image_classification_with_densenet/image_cnn_train/image_classification/training.py b/cli/jobs/pipelines-with-components/image_classification_with_densenet/image_cnn_train/image_classification/training.py
index f8f6f170ff..578fa440e4 100644
--- a/cli/jobs/pipelines-with-components/image_classification_with_densenet/image_cnn_train/image_classification/training.py
+++ b/cli/jobs/pipelines-with-components/image_classification_with_densenet/image_cnn_train/image_classification/training.py
@@ -623,7 +623,7 @@ def train_loop(
         ts = str(time.time())
         # logdir = os.path.expanduser('~/tensorboard/{}/logs/'.format(os.environ['DLTS_JOB_ID']) + ts)
         logdir = os.path.expanduser(
-            "~/tensorboard/{}/logs/".format(os.environ["AZ_BATCH_JOB_ID"]) + ts
+            "~/tensorboard/{}/logs/".format(os.environ["AZUREML_RUN_ID"]) + ts
         )
         print("tensorboard at ", logdir)
         if not os.path.exists(logdir):
diff --git a/sdk/python/endpoints/online/kubernetes/kubernetes-online-endpoints-simple-deployment.ipynb b/sdk/python/endpoints/online/kubernetes/kubernetes-online-endpoints-simple-deployment.ipynb
index 69cefc9fc6..14ea25d210 100644
--- a/sdk/python/endpoints/online/kubernetes/kubernetes-online-endpoints-simple-deployment.ipynb
+++ b/sdk/python/endpoints/online/kubernetes/kubernetes-online-endpoints-simple-deployment.ipynb
@@ -144,6 +144,7 @@
    "source": [
     "# Install docker package in the current Jupyter kernel\n",
     "import sys\n",
+    "\n",
     "!{sys.executable} -m pip install docker"
    ]
   },