diff --git a/src/index.ts b/src/index.ts
index 3921ec1..840d9fb 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -84,7 +84,8 @@ const plugin: JupyterFrontEndPlugin<void> = {
       app,
       kernelsRegistry,
       commsRegistry,
-      init
+      init,
+      updateOutputs
     };
 
     (window as any).trameJupyter = namespace;
diff --git a/src/utils.ts b/src/utils.ts
index 24850bb..e81f6d1 100644
--- a/src/utils.ts
+++ b/src/utils.ts
@@ -1,8 +1,10 @@
 export function updateOutputs() {
   const full = document.querySelectorAll<HTMLElement>(
-    '.jp-LinkedOutputView iframe[id^=trame]'
+    '.jp-LinkedOutputView iframe[id*=trame__template]'
+  );
+  const all = document.querySelectorAll<HTMLElement>(
+    'iframe[id*=trame__template]'
   );
-  const all = document.querySelectorAll<HTMLElement>('iframe[id^=trame]');
   const visibleInFullScreen: Record<string, boolean> = {};
   const moveToFullScreen: Array<HTMLElement> = [];