diff --git a/test/behavior/sniff.test.js b/test/behavior/sniff.test.js
index 22b84e2e2..bf0a7b386 100644
--- a/test/behavior/sniff.test.js
+++ b/test/behavior/sniff.test.js
@@ -45,7 +45,8 @@ test('Should update the connection pool', t => {
             roles: {
               master: true,
               data: true,
-              ingest: true
+              ingest: true,
+              ml: false
             }
           })
         } else {
@@ -55,7 +56,8 @@ test('Should update the connection pool', t => {
             roles: {
               master: true,
               data: true,
-              ingest: true
+              ingest: true,
+              ml: false
             },
             ssl: null,
             agent: null
diff --git a/test/unit/client.test.js b/test/unit/client.test.js
index 49a71e3fe..53510ba6a 100644
--- a/test/unit/client.test.js
+++ b/test/unit/client.test.js
@@ -21,8 +21,7 @@ test('Configure host', t => {
         master: true,
         data: true,
         ingest: true,
-        coordinating: true,
-        machine_learning: true
+        ml: false
       }
     })
     t.end()
@@ -43,8 +42,7 @@ test('Configure host', t => {
         master: true,
         data: true,
         ingest: true,
-        coordinating: true,
-        machine_learning: true
+        ml: false
       }
     })
     t.match(pool.connections.get('http://localhost:9201/'), {
@@ -57,8 +55,7 @@ test('Configure host', t => {
         master: true,
         data: true,
         ingest: true,
-        coordinating: true,
-        machine_learning: true
+        ml: false
       }
     })
 
@@ -72,7 +69,8 @@ test('Configure host', t => {
         id: 'node',
         roles: {
           master: true,
-          data: false
+          data: false,
+          ingest: false
         },
         ssl: 'ssl'
       }
@@ -83,12 +81,16 @@ test('Configure host', t => {
       id: 'node',
       ssl: 'ssl',
       deadCount: 0,
-      resurrectTimeout: 0,
-      roles: {
-        master: true,
-        data: false
-      }
+      resurrectTimeout: 0
     })
+
+    t.deepEqual(pool.connections.get('node').roles, {
+      master: true,
+      data: false,
+      ingest: false,
+      ml: false
+    })
+
     t.end()
   })
 
@@ -99,7 +101,8 @@ test('Configure host', t => {
         id: 'node1',
         roles: {
           master: true,
-          data: false
+          data: false,
+          ingest: false
         },
         ssl: 'ssl'
       }, {
@@ -107,7 +110,8 @@ test('Configure host', t => {
         id: 'node2',
         roles: {
           master: false,
-          data: true
+          data: true,
+          ingest: false
         },
         ssl: 'ssl'
       }]
@@ -118,23 +122,31 @@ test('Configure host', t => {
       id: 'node1',
       ssl: 'ssl',
       deadCount: 0,
-      resurrectTimeout: 0,
-      roles: {
-        master: true,
-        data: false
-      }
+      resurrectTimeout: 0
     })
+
+    t.deepEqual(pool.connections.get('node1').roles, {
+      master: true,
+      data: false,
+      ingest: false,
+      ml: false
+    })
+
     t.match(pool.connections.get('node2'), {
       url: new URL('http://localhost:9200'),
       id: 'node2',
       ssl: 'ssl',
       deadCount: 0,
-      resurrectTimeout: 0,
-      roles: {
-        master: false,
-        data: true
-      }
+      resurrectTimeout: 0
     })
+
+    t.deepEqual(pool.connections.get('node2').roles, {
+      master: false,
+      data: true,
+      ingest: false,
+      ml: false
+    })
+
     t.end()
   })
 
diff --git a/test/unit/connection-pool.test.js b/test/unit/connection-pool.test.js
index 5ca572ffe..cf054aade 100644
--- a/test/unit/connection-pool.test.js
+++ b/test/unit/connection-pool.test.js
@@ -296,7 +296,8 @@ test('API', t => {
       roles: {
         master: true,
         data: true,
-        ingest: true
+        ingest: true,
+        ml: false
       }
     }, {
       url: new URL('http://127.0.0.1:9201'),
@@ -304,7 +305,8 @@ test('API', t => {
       roles: {
         master: true,
         data: true,
-        ingest: true
+        ingest: true,
+        ml: false
       }
     }])
     t.end()
@@ -426,7 +428,8 @@ test('API', t => {
       t.deepEqual(pool.connections.get('a1').roles, {
         master: true,
         data: true,
-        ingest: true
+        ingest: true,
+        ml: false
       })
       t.strictEqual(pool.connections.get('http://127.0.0.1:9200/'), undefined)
     })