diff --git a/template/v2/dirs/usr/local/bin/start-jupyter-server b/template/v2/dirs/usr/local/bin/start-jupyter-server
index 7b20d386..c9aedde6 100755
--- a/template/v2/dirs/usr/local/bin/start-jupyter-server
+++ b/template/v2/dirs/usr/local/bin/start-jupyter-server
@@ -10,7 +10,7 @@ if [ -n "$SAGEMAKER_RECOVERY_MODE" ]; then
 else
     # Activate conda environment 'base'
     micromamba activate base
-    jupyter labextension disable @amzn/sagemaker-data-explorer-jl-plugin sagemaker-data-explorer:plugin amzn/sagemaker-ui-theme-jlplugin @amzn/sagemaker-ui-doc-manager-jl-plugin @amzn/sagemaker-connection-magics-jlextension
+    jupyter labextension disable @amzn/sagemaker-data-explorer-jl-plugin sagemaker-data-explorer:plugin amzn/sagemaker-ui-theme-jlplugin @amzn/sagemaker-ui-doc-manager-jl-plugin @amzn/sagemaker-connection-magics-jlextension sagemaker_gen_ai_jupyterlab_extension
 fi
 
 # Start Jupyter server in rtc mode for shared spaces
diff --git a/template/v2/dirs/usr/local/bin/start-sagemaker-ui-jupyter-server b/template/v2/dirs/usr/local/bin/start-sagemaker-ui-jupyter-server
index 3c03c50a..0cd986b5 100755
--- a/template/v2/dirs/usr/local/bin/start-sagemaker-ui-jupyter-server
+++ b/template/v2/dirs/usr/local/bin/start-sagemaker-ui-jupyter-server
@@ -10,6 +10,9 @@ if [ -n "$SAGEMAKER_RECOVERY_MODE" ]; then
 else
     # Activate conda environment 'base' which is the default for Cosmos
     micromamba activate base
+
+    # Disable jupyter-ai in favor of sagemaker_gen_ai_jupyterlab_extension
+    jupyter labextension disable @jupyter-ai/core
 fi
 
 sudo cp -r /etc/sagemaker-ui/kernels/. /opt/conda/share/jupyter/kernels/
diff --git a/template/v3/dirs/usr/local/bin/start-jupyter-server b/template/v3/dirs/usr/local/bin/start-jupyter-server
index a0242ffb..57535810 100755
--- a/template/v3/dirs/usr/local/bin/start-jupyter-server
+++ b/template/v3/dirs/usr/local/bin/start-jupyter-server
@@ -10,7 +10,7 @@ if [ -n "$SAGEMAKER_RECOVERY_MODE" ]; then
 else
     # Activate conda environment 'base'
     micromamba activate base
-    jupyter labextension disable @amzn/sagemaker-data-explorer-jl-plugin sagemaker-data-explorer:plugin amzn/sagemaker-ui-theme-jlplugin @amzn/sagemaker-ui-doc-manager-jl-plugin @amzn/sagemaker-connection-magics-jlextension
+    jupyter labextension disable @amzn/sagemaker-data-explorer-jl-plugin sagemaker-data-explorer:plugin amzn/sagemaker-ui-theme-jlplugin @amzn/sagemaker-ui-doc-manager-jl-plugin @amzn/sagemaker-connection-magics-jlextension sagemaker_gen_ai_jupyterlab_extension
 fi
 
 # Start Jupyter server in rtc mode for shared spaces
diff --git a/template/v3/dirs/usr/local/bin/start-sagemaker-ui-jupyter-server b/template/v3/dirs/usr/local/bin/start-sagemaker-ui-jupyter-server
index dfe780ef..a2e7cb25 100755
--- a/template/v3/dirs/usr/local/bin/start-sagemaker-ui-jupyter-server
+++ b/template/v3/dirs/usr/local/bin/start-sagemaker-ui-jupyter-server
@@ -10,6 +10,9 @@ if [ -n "$SAGEMAKER_RECOVERY_MODE" ]; then
 else
     # Activate conda environment 'base' which is the default for Cosmos
     micromamba activate base
+
+    # Disable jupyter-ai in favor of sagemaker_gen_ai_jupyterlab_extension
+    jupyter labextension disable @jupyter-ai/core
 fi
 
 sudo cp -r /etc/sagemaker-ui/kernels/. /opt/conda/share/jupyter/kernels/