diff --git a/src/public/app/widgets/floating_buttons/relation_map_buttons.js b/src/public/app/widgets/floating_buttons/relation_map_buttons.js
index 72cdb7bf3..5940b67b2 100644
--- a/src/public/app/widgets/floating_buttons/relation_map_buttons.js
+++ b/src/public/app/widgets/floating_buttons/relation_map_buttons.js
@@ -1,7 +1,4 @@
 import NoteContextAwareWidget from "../note_context_aware_widget.js";
-import dialogService from "../dialog.js";
-import server from "../../services/server.js";
-import toastService from "../../services/toast.js";
 
 const TPL = `
 <div>
diff --git a/src/public/app/widgets/note_wrapper.js b/src/public/app/widgets/note_wrapper.js
index cae922a3a..43979ada4 100644
--- a/src/public/app/widgets/note_wrapper.js
+++ b/src/public/app/widgets/note_wrapper.js
@@ -29,14 +29,19 @@ export default class NoteWrapperWidget extends FlexContainer {
     }
 
     refresh() {
+        const isHiddenExt = this.isHiddenExt(); // preserve through class reset
+
         this.$widget.removeClass();
 
+        this.toggleExt(!isHiddenExt);
+
+        this.$widget.addClass("component note-split");
+
         const note = this.noteContext?.note;
         if (!note) {
             return;
         }
 
-        this.$widget.addClass("note-split");
         this.$widget.toggleClass("full-content-width",
             ['image', 'mermaid', 'book', 'render', 'canvas', 'web-view'].includes(note.type)
             || !!note?.hasLabel('fullContentWidth')