diff --git a/src/GraphWorkspace.jsx b/src/GraphWorkspace.jsx
index cade4bd..c53a5a0 100644
--- a/src/GraphWorkspace.jsx
+++ b/src/GraphWorkspace.jsx
@@ -17,9 +17,9 @@ export default class GraphWorkspace extends React.PureComponent {
         },
     };
 
-    componentDidUpdate(prevProps, prevState) {
-        if (this.props.graph != prevProps.graph) {
-            this.setState({selectedNodes:[]})
+    componentWillReceiveProps(nextProps) {
+        if (nextProps.graph !== this.props.graph) {
+            this.setState({selectedNodes:[]});
         }
     }