diff --git a/automl/beta/src/test/java/com/example/automl/BatchPredictTest.java b/automl/beta/src/test/java/com/example/automl/BatchPredictTest.java
index 688c6702c00..bcba7637ecd 100644
--- a/automl/beta/src/test/java/com/example/automl/BatchPredictTest.java
+++ b/automl/beta/src/test/java/com/example/automl/BatchPredictTest.java
@@ -74,10 +74,10 @@ public void testBatchPredict() {
       BatchPredict.batchPredict(PROJECT_ID, MODEL_ID, inputUri, outputUri);
       String got = bout.toString();
       assertThat(got)
-          .contains("The model is either not found or not supported for prediction yet.");
+          .contains("does not exist");
     } catch (IOException | ExecutionException | InterruptedException e) {
       assertThat(e.getMessage())
-          .contains("The model is either not found or not supported for prediction yet.");
+          .contains("does not exist");
     }
   }
 }
diff --git a/automl/cloud-client/src/test/java/com/example/automl/BatchPredictTest.java b/automl/cloud-client/src/test/java/com/example/automl/BatchPredictTest.java
index d46b669ac1d..974caaa6e7d 100644
--- a/automl/cloud-client/src/test/java/com/example/automl/BatchPredictTest.java
+++ b/automl/cloud-client/src/test/java/com/example/automl/BatchPredictTest.java
@@ -83,10 +83,10 @@ public void testBatchPredict() {
       BatchPredict.batchPredict(PROJECT_ID, MODEL_ID, inputUri, outputUri);
       String got = bout.toString();
       assertThat(got)
-          .contains("The model is either not found or not supported for prediction yet.");
+          .contains("does not exist");
     } catch (IOException | ExecutionException | InterruptedException e) {
       assertThat(e.getMessage())
-          .contains("The model is either not found or not supported for prediction yet.");
+          .contains("does not exist");
     }
   }
 }