diff --git a/src/components/content.js b/src/components/content.js
index 6ee26a38d..2d441a189 100644
--- a/src/components/content.js
+++ b/src/components/content.js
@@ -880,6 +880,7 @@ class Content extends React.Component {
         data-slate-editor
         key={this.tmp.forces}
         ref={this.ref}
+        data-key={document.key}
         contentEditable={!readOnly}
         suppressContentEditableWarning
         className={className}
diff --git a/test/rendering/index.js b/test/rendering/index.js
index 550307ea3..1b0108698 100644
--- a/test/rendering/index.js
+++ b/test/rendering/index.js
@@ -59,6 +59,7 @@ function clean(html) {
   $.root().children().removeAttr('autocorrect')
   $.root().children().removeAttr('spellcheck')
   $.root().children().removeAttr('style')
+  $.root().children().removeAttr('data-gramm')
 
   return $.html()
 }