diff --git a/addon/chrome/content/workspaceWindow.xhtml b/addon/chrome/content/workspaceWindow.xhtml
index c7cdcc2..f793955 100644
--- a/addon/chrome/content/workspaceWindow.xhtml
+++ b/addon/chrome/content/workspaceWindow.xhtml
@@ -37,6 +37,24 @@
             "chrome://global/content/customElements.js",
             this
           );
+          if (!customElements.get("note-editor")) {
+            Services.scriptloader.loadSubScript(
+              "chrome://zotero/content/elements/shadowAutocompleteInput.js",
+              this
+            );
+            Services.scriptloader.loadSubScript(
+              "chrome://zotero/content/elements/noteEditor.js",
+              this
+            );
+            Services.scriptloader.loadSubScript(
+              "chrome://zotero/content/elements/relatedBox.js",
+              this
+            );
+            Services.scriptloader.loadSubScript(
+              "chrome://zotero/content/elements/tagsBox.js",
+              this
+            );
+          }
         }
         window.arguments[0]._initPromise.resolve();
       });
diff --git a/src/modules/workspace/content.ts b/src/modules/workspace/content.ts
index d78a384..447cfcb 100644
--- a/src/modules/workspace/content.ts
+++ b/src/modules/workspace/content.ts
@@ -223,7 +223,9 @@ export function initWorkspace(container: XUL.Box | undefined) {
   );
   // Manually add custom editor items in Zotero 7
   if (ztoolkit.isZotero7()) {
-    const customElements = ztoolkit.getGlobal("customElements");
+    // @ts-ignore
+    const customElements = container.ownerGlobal
+      .customElements as CustomElementRegistry;
     const mainEditorContainer = container.querySelector(
       `#${makeId("editor-main-container")}`
     );