diff --git a/images/jupyter-ai.yml b/images/jupyter-ai.yml
index 1a6f808..3a7aca9 100644
--- a/images/jupyter-ai.yml
+++ b/images/jupyter-ai.yml
@@ -4,7 +4,8 @@ channels:
 dependencies:
   - jupyterlab-myst
   - jupyter-ai
-  - jupyter-tensorboard-proxy 
   - jupyter-resource-usage
   - jupyter-vscode-proxy
+  - pip:
+    - jupyter-tensorboard-proxy