From 21afabfde715da76fe472fecdc6f9938f2635b3b Mon Sep 17 00:00:00 2001
From: zane-neo <zaniu@amazon.com>
Date: Wed, 7 Feb 2024 17:53:20 +0800
Subject: [PATCH] Fix failure UTs

Signed-off-by: zane-neo <zaniu@amazon.com>
---
 .../ml/action/models/DeleteModelTransportActionTests.java       | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/plugin/src/test/java/org/opensearch/ml/action/models/DeleteModelTransportActionTests.java b/plugin/src/test/java/org/opensearch/ml/action/models/DeleteModelTransportActionTests.java
index 37c6f8755f..bfc7f628ff 100644
--- a/plugin/src/test/java/org/opensearch/ml/action/models/DeleteModelTransportActionTests.java
+++ b/plugin/src/test/java/org/opensearch/ml/action/models/DeleteModelTransportActionTests.java
@@ -212,7 +212,7 @@ public void testDeleteRemoteModel_deleteModelController_failed() throws IOExcept
         deleteModelTransportAction.doExecute(null, mlModelDeleteRequest, actionListener);
         ArgumentCaptor<Exception> argumentCaptor = ArgumentCaptor.forClass(Exception.class);
         verify(actionListener).onFailure(argumentCaptor.capture());
-        assertEquals("Failed to delete model controller, please try again: test_id", argumentCaptor.getValue().getMessage());
+        assertEquals("Model is not all cleaned up, please try again: test_id", argumentCaptor.getValue().getMessage());
     }
 
     public void testDeleteLocalModel_deleteModelController_failed() throws IOException {