diff --git a/packages/slate/src/changes/with-schema.js b/packages/slate/src/changes/with-schema.js
index ecd726c59..5d44a8772 100644
--- a/packages/slate/src/changes/with-schema.js
+++ b/packages/slate/src/changes/with-schema.js
@@ -45,7 +45,7 @@ Changes.normalizeNodeByKey = (change, key) => {
 
   normalizeNodeAndChildren(change, node, schema)
 
-  document = change.state.document
+  document = change.value.document
   const ancestors = document.getAncestors(key)
   if (!ancestors) return