diff --git a/sdks/java/core/src/test/java/org/apache/beam/sdk/fn/splittabledofn/RestrictionTrackersTest.java b/sdks/java/core/src/test/java/org/apache/beam/sdk/fn/splittabledofn/RestrictionTrackersTest.java
index 8f7ee9eb25dd..7b6f3d47c273 100644
--- a/sdks/java/core/src/test/java/org/apache/beam/sdk/fn/splittabledofn/RestrictionTrackersTest.java
+++ b/sdks/java/core/src/test/java/org/apache/beam/sdk/fn/splittabledofn/RestrictionTrackersTest.java
@@ -102,7 +102,7 @@ private static class RestrictionTrackerWithProgress extends RestrictionTracker<O
       implements HasProgress {
     private boolean blockTryClaim;
     private boolean blockTrySplit;
-    private boolean isBlocked;
+    private volatile boolean isBlocked;
     public static final Progress REPORT_PROGRESS = Progress.from(2.0, 3.0);
 
     public RestrictionTrackerWithProgress() {