diff --git a/packages/idyll-document/src/index.js b/packages/idyll-document/src/index.js
index 48d2a5252..a5db23d8c 100644
--- a/packages/idyll-document/src/index.js
+++ b/packages/idyll-document/src/index.js
@@ -90,6 +90,10 @@ class IdyllDocument extends React.Component {
     }
 
     if (newProps.ast) {
+      this.setState({
+        previousAST: this.state.ast,
+        hash: hashCode(newProps.ast.toString())
+      });
       return;
     }