[ML] address PR comments

elastic · Feb 18, 2020 · e60e183 · e60e183
1 parent cc6f889
commit e60e183
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/x-pack/legacy/plugins/ml/server/routes/system.ts b/x-pack/legacy/plugins/ml/server/routes/system.ts
@@ -27,7 +27,7 @@ export function systemRoutes({
 }: RouteInitialization) {
   async function getNodeCount(context: RequestHandlerContext) {
     const filterPath = 'nodes.*.attributes';
-    const resp = await context.core.elasticsearch.adminClient.callAsInternalUser('nodes.info', {
+    const resp = await context.ml!.mlClient.callAsInternalUser('nodes.info', {
       filterPath,
     });
 
@@ -176,7 +176,7 @@ export function systemRoutes({
           const body = { cluster: requiredPrivileges };
           const resp = await context.ml!.mlClient.callAsCurrentUser('ml.privilegeCheck', { body });
 
-          if (requiredPrivileges.every(p => !!resp.cluster[p])) {
+          if (resp.has_all_requested) {
             return response.ok({
               body: await getNodeCount(context),
             });