diff --git a/java-vision/google-cloud-vision/src/test/java/com/google/cloud/vision/it/ITSystemTest.java b/java-vision/google-cloud-vision/src/test/java/com/google/cloud/vision/it/ITSystemTest.java
index a264f37d40fa..075fadd1497b 100644
--- a/java-vision/google-cloud-vision/src/test/java/com/google/cloud/vision/it/ITSystemTest.java
+++ b/java-vision/google-cloud-vision/src/test/java/com/google/cloud/vision/it/ITSystemTest.java
@@ -292,7 +292,8 @@ public void detectFacesTest() throws IOException {
     for (FaceAnnotation annotation : assertNotEmpty(res, res.getFaceAnnotationsList())) {
       assertThat(annotation.getAngerLikelihood()).isAnyOf(Likelihood.LIKELY, Likelihood.POSSIBLE);
       assertEquals(Likelihood.VERY_UNLIKELY, annotation.getJoyLikelihood());
-      assertEquals(Likelihood.LIKELY, annotation.getSurpriseLikelihood());
+      assertThat(annotation.getSurpriseLikelihood())
+          .isAnyOf(Likelihood.LIKELY, Likelihood.POSSIBLE);
     }
   }
 
@@ -303,7 +304,8 @@ public void detectFacesGcsTest() throws IOException {
     for (FaceAnnotation annotation : assertNotEmpty(res, res.getFaceAnnotationsList())) {
       assertThat(annotation.getAngerLikelihood()).isAnyOf(Likelihood.LIKELY, Likelihood.POSSIBLE);
       assertEquals(Likelihood.VERY_UNLIKELY, annotation.getJoyLikelihood());
-      assertEquals(Likelihood.LIKELY, annotation.getSurpriseLikelihood());
+      assertThat(annotation.getSurpriseLikelihood())
+          .isAnyOf(Likelihood.LIKELY, Likelihood.POSSIBLE);
     }
   }