Fix ZacharyRSmith#42: moved unblock assign to CDM

kuhelbeher · Jan 13, 2019 · 2fe55eb · 2fe55eb
1 parent 28582f1
commit 2fe55eb
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/index.js b/src/index.js
@@ -69,13 +69,15 @@ class NavigationPrompt extends React.Component<PropsT, StateT> {
     (this:Object).onConfirm = this.onConfirm.bind(this);
     (this:Object).when = this.when.bind(this);
 
-    this.state = {...initState, unblock: props.history.block(this.block)};
+    this.state = {...initState};
   }
 
   componentDidMount() {
     if (!this.props.disableNative) {
       window.addEventListener('beforeunload', this.onBeforeUnload);
     }
+
+    this.setState({unblock: this.props.history.block(this.block)});
   }
 
   componentDidUpdate(prevProps, prevState) {