Fix

PostHog · Jan 2, 2024 · af63ae8 · af63ae8
1 parent c7d3138
commit af63ae8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/frontend/src/scenes/notebooks/Notebook/notebookLogic.ts b/frontend/src/scenes/notebooks/Notebook/notebookLogic.ts
@@ -75,7 +75,7 @@ export const notebookLogic = kea<notebookLogicType>([
                 scope: ActivityScope.NOTEBOOK,
                 item_id: props.shortId,
             }),
-            ['comments'],
+            ['comments', 'itemContext'],
         ],
         actions: [
             notebooksModel,
@@ -86,7 +86,7 @@ export const notebookLogic = kea<notebookLogicType>([
                 scope: ActivityScope.NOTEBOOK,
                 item_id: props.shortId,
             }),
-            ['setItemContext', 'sendComposedContentSuccess', 'maybeLoadComments'],
+            ['setItemContext', 'maybeLoadComments'],
         ],
     })),
     actions({
@@ -669,7 +669,7 @@ export const notebookLogic = kea<notebookLogicType>([
                     .map((comment) => comment.item_context?.id)
 
                 editor.getMarks('comment').forEach((mark) => {
-                    if (!commentMarkIds.includes(mark.id)) {
+                    if (!commentMarkIds.includes(mark.id) && values.itemContext?.context?.id !== mark.id) {
                         editor.removeComment(mark.pos)
                     }
                 })