diff --git a/runners/google-cloud-dataflow-java/worker/src/test/java/org/apache/beam/runners/dataflow/worker/windmill/state/WindmillTagEncodingV2Test.java b/runners/google-cloud-dataflow-java/worker/src/test/java/org/apache/beam/runners/dataflow/worker/windmill/state/WindmillTagEncodingV2Test.java
index af9ef95410d1..d06b4e9ccbad 100644
--- a/runners/google-cloud-dataflow-java/worker/src/test/java/org/apache/beam/runners/dataflow/worker/windmill/state/WindmillTagEncodingV2Test.java
+++ b/runners/google-cloud-dataflow-java/worker/src/test/java/org/apache/beam/runners/dataflow/worker/windmill/state/WindmillTagEncodingV2Test.java
@@ -319,7 +319,13 @@ public void testTimerTag() {
                   new Instant(123),
                   new Instant(456),
                   timeDomain)
-              : TimerData.of(TIMER_ID, namespace, new Instant(123), new Instant(456), timeDomain);
+              : TimerData.of(
+                  TIMER_ID,
+                  namespace,
+                  new Instant(123),
+                  new Instant(456),
+                  timeDomain,
+                  TimerData.CausedByDrain.NORMAL);
       assertEquals(expectedBytes, WindmillTagEncodingV2.instance().timerTag(prefix, timerData));
     }
   }