better fix

Author: jan-elastic

x-pack/plugin/ml/src/main/java/org/elasticsearch/xpack/ml/action/TransportGetDeploymentStatsAction.java

@@ -231,7 +231,7 @@ static GetDeploymentStatsAction.Response addFailedRoutes(
                     }
                 }
 
-                updatedNodeStats.sort(Comparator.comparing(n -> n.getNode() != null ? n.getNode().getId() : ""));
+                updatedNodeStats.sort(Comparator.comparing(n -> n.getNode().getId()));
                 updatedAssignmentStats.add(
                     new AssignmentStats(
                         stat.getDeploymentId(),
@@ -261,16 +261,18 @@ static GetDeploymentStatsAction.Response addFailedRoutes(
                 List<AssignmentStats.NodeStats> nodeStats = new ArrayList<>();
 
                 for (var routingEntry : nonStartedEntries.getValue().entrySet()) {
-                    nodeStats.add(
-                        AssignmentStats.NodeStats.forNotStartedState(
-                            nodes.get(routingEntry.getKey()),
-                            routingEntry.getValue().getState(),
-                            routingEntry.getValue().getReason()
-                        )
-                    );
+                    if (nodes.nodeExists(routingEntry.getKey())) {
+                        nodeStats.add(
+                            AssignmentStats.NodeStats.forNotStartedState(
+                                nodes.get(routingEntry.getKey()),
+                                routingEntry.getValue().getState(),
+                                routingEntry.getValue().getReason()
+                            )
+                        );
+                    }
                 }
 
-                nodeStats.sort(Comparator.comparing(n -> n.getNode() != null ? n.getNode().getId() : ""));
+                nodeStats.sort(Comparator.comparing(n -> n.getNode().getId()));
 
                 updatedAssignmentStats.add(
                     new AssignmentStats(