diff --git a/java-spanner/google-cloud-spanner/src/test/java/com/google/cloud/spanner/it/ITBatchReadTest.java b/java-spanner/google-cloud-spanner/src/test/java/com/google/cloud/spanner/it/ITBatchReadTest.java
index c7b43635554c..ee95454ee63b 100644
--- a/java-spanner/google-cloud-spanner/src/test/java/com/google/cloud/spanner/it/ITBatchReadTest.java
+++ b/java-spanner/google-cloud-spanner/src/test/java/com/google/cloud/spanner/it/ITBatchReadTest.java
@@ -250,7 +250,7 @@ public void readUsingIndex() {
   @Test
   public void dataBoostRead() {
     assumeFalse("data boost is not supported on Spanner Omni yet", isSpannerOmni());
-    assumeFalse("Skipping the test due to a known bug b/557058207", env.isCloudDevelOrStaging());
+    assumeFalse("Skipping the test due to a known bug b/557058207", env.isCloudStaging());
     BitSet seenRows = new BitSet(numRows);
     TimestampBound bound = getRandomBound();
     PartitionOptions partitionParams = getRandomPartitionOptions();
@@ -304,7 +304,7 @@ private PartitionOptions getRandomPartitionOptions() {
   @Test
   public void dataBoostQuery() {
     assumeFalse("data boost is not supported on Spanner Omni yet", isSpannerOmni());
-    assumeFalse("Skipping the test due to a known bug b/557058207", env.isCloudDevelOrStaging());
+    assumeFalse("Skipping the test due to a known bug b/557058207", env.isCloudStaging());
     BitSet seenRows = new BitSet(numRows);
     TimestampBound bound = getRandomBound();
     PartitionOptions partitionParams = getRandomPartitionOptions();