diff --git a/code_search/code-search.ipynb b/code_search/code-search.ipynb
index 4e7d13af..f67b3d72 100644
--- a/code_search/code-search.ipynb
+++ b/code_search/code-search.ipynb
@@ -198,6 +198,9 @@
     "# Update Ksonnet to point to the Kubernetes Cluster\n",
     "ks env add code-search --context $(kubectl config current-context)\n",
     "\n",
+    "# Update Ksonnet to use the namespace where kubeflow is deployed. By default it's 'kubeflow'\n",
+    "ks env set code-search --namespace=kubeflow\n",
+    "\n",
     "# Update the Working Directory of the application\n",
     "sed -i'' \"s,gs://example/prefix,${WORKING_DIR},\" components/params.libsonnet\n",
     "\n",