diff --git a/lit_nlp/notebook.py b/lit_nlp/notebook.py
index aae2eb9a..02ad0e4e 100644
--- a/lit_nlp/notebook.py
+++ b/lit_nlp/notebook.py
@@ -51,7 +51,6 @@
             *layout.MODEL_PREDS_MODULES,
             modules.SalienceMapModule,
             modules.LegacySequenceSalienceModule,
-            modules.AttentionModule,
         ],
         'Analysis': [
             modules.MetricsModule,