diff --git a/extensions/notebook/src/jupyter/jupyterSessionManager.ts b/extensions/notebook/src/jupyter/jupyterSessionManager.ts
index 3cdd97e9da..fabd6ebd3f 100644
--- a/extensions/notebook/src/jupyter/jupyterSessionManager.ts
+++ b/extensions/notebook/src/jupyter/jupyterSessionManager.ts
@@ -351,6 +351,9 @@ export class JupyterSession implements nb.ISession {
 				}
 			}
 
+			// To set ADS context for Kqlmagic
+			allCode += `%set_env KQLMAGIC_NOTEBOOK_APP=AzureDataStudio${EOL}`;
+
 			// Enable silent mode for perf improvement of reload command.
 			allCode += `%set_env KQLMAGIC_LOAD_MODE=silent${EOL}`;