From aecee236b4520fe7e1caaaa67e9bccdf7dba9a39 Mon Sep 17 00:00:00 2001
From: David Roberts <dave.roberts@elastic.co>
Date: Mon, 6 Sep 2021 11:05:32 +0100
Subject: [PATCH] [ML] Fix typo in ML shutdown test

There was a small typo in #77281 that made the test fail 2/9ths
of the time.

Fixes #77297
---
 .../elasticsearch/xpack/ml/integration/MlNodeShutdownIT.java   | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/x-pack/plugin/ml/src/internalClusterTest/java/org/elasticsearch/xpack/ml/integration/MlNodeShutdownIT.java b/x-pack/plugin/ml/src/internalClusterTest/java/org/elasticsearch/xpack/ml/integration/MlNodeShutdownIT.java
index 018bd5b309205..b16064f639b1c 100644
--- a/x-pack/plugin/ml/src/internalClusterTest/java/org/elasticsearch/xpack/ml/integration/MlNodeShutdownIT.java
+++ b/x-pack/plugin/ml/src/internalClusterTest/java/org/elasticsearch/xpack/ml/integration/MlNodeShutdownIT.java
@@ -34,7 +34,6 @@
 
 public class MlNodeShutdownIT extends BaseMlIntegTestCase {
 
-    @AwaitsFix(bugUrl="https://github.com/elastic/elasticsearch/issues/77297")
     public void testJobsVacateShuttingDownNode() throws Exception {
 
         internalCluster().ensureAtLeastNumDataNodes(3);
@@ -81,7 +80,7 @@ public void testJobsVacateShuttingDownNode() throws Exception {
             PutShutdownNodeAction.INSTANCE,
             new PutShutdownNodeAction.Request(
                 nodeIdToShutdown.get(),
-                randomFrom(SingleNodeShutdownMetadata.Type.values()),
+                type,
                 "just testing",
                 null,
                 targetNodeName)