diff --git a/wax-prosemirror-core/src/Wax.js b/wax-prosemirror-core/src/Wax.js
index a312359990fbdb3b600a3935b35f36a44373c327..23d6140129d7c9dea045477560a8c153a107ea44 100644
--- a/wax-prosemirror-core/src/Wax.js
+++ b/wax-prosemirror-core/src/Wax.js
@@ -61,7 +61,7 @@ const Wax = props => {
 
     if (notes.length > 0) {
       notes.forEach(note => {
-        schema.nodes.footnote.spec.toDOM = node => {
+        schema.nodes[note.name].spec.toDOM = node => {
           // eslint-disable-next-line prefer-rest-params
           if (node) return [note.name, node.attrs, 0];
         };
@@ -77,7 +77,7 @@ const Wax = props => {
 
     if (notes.length > 0) {
       notes.forEach(note => {
-        schema.nodes.footnote.spec.toDOM = node => {
+        schema.nodes[note.name].spec.toDOM = node => {
           // eslint-disable-next-line prefer-rest-params
           if (node) return [note.name, node.attrs];
         };