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 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 {