diff --git a/client/app/scripts/charts/nodes-chart.js b/client/app/scripts/charts/nodes-chart.js
index 2a6d8b674..64380771a 100644
--- a/client/app/scripts/charts/nodes-chart.js
+++ b/client/app/scripts/charts/nodes-chart.js
@@ -50,6 +50,12 @@ const NodesChart = React.createClass({
});
this.updateGraphState(nextProps);
}
+ if (this.props.selectedNodeId !== nextProps.selectedNodeId) {
+ this.restoreLayout();
+ }
+ if (nextProps.selectedNodeId) {
+ this.centerSelectedNode(nextProps);
+ }
},
componentWillUnmount: function() {
@@ -77,7 +83,8 @@ const NodesChart = React.createClass({
renderGraphNodes: function(nodes, scale) {
return _.map(nodes, function(node) {
- const highlighted = _.includes(this.props.highlightedNodeIds, node.id);
+ const highlighted = _.includes(this.props.highlightedNodeIds, node.id)
+ || this.props.selectedNodeId === node.id;
return (