From 23ed85325cd827ced1361cbf3e7bee8a8f3f5960 Mon Sep 17 00:00:00 2001
From: Shunping Huang <shunping@google.com>
Date: Mon, 3 Aug 2026 10:28:03 -0400
Subject: [PATCH] Fix internal test failure after #39487

---
 .../dataflow/worker/WindmillKeyedWorkItem.java       | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/runners/google-cloud-dataflow-java/worker/src/main/java/org/apache/beam/runners/dataflow/worker/WindmillKeyedWorkItem.java b/runners/google-cloud-dataflow-java/worker/src/main/java/org/apache/beam/runners/dataflow/worker/WindmillKeyedWorkItem.java
index ff5be071edea..290c25836f63 100644
--- a/runners/google-cloud-dataflow-java/worker/src/main/java/org/apache/beam/runners/dataflow/worker/WindmillKeyedWorkItem.java
+++ b/runners/google-cloud-dataflow-java/worker/src/main/java/org/apache/beam/runners/dataflow/worker/WindmillKeyedWorkItem.java
@@ -201,12 +201,14 @@ public Iterable<TimerData> timersIterable() {
   }
 
   @Override
-  @SuppressWarnings("nullness")
+  @SuppressWarnings({"nullness", "unchecked"})
   public Iterable<WindowedValue<?>> elementWindowsIterable() {
-    return FluentIterable.from(workItem.getMessageBundlesList())
-        .transformAndConcat(Windmill.InputMessageBundle::getMessagesList)
-        .transform(this::parseElemWindowOnly)
-        .filter(Objects::nonNull);
+    return (Iterable<WindowedValue<?>>)
+        (Iterable<?>)
+            FluentIterable.from(workItem.getMessageBundlesList())
+                .transformAndConcat(Windmill.InputMessageBundle::getMessagesList)
+                .transform(this::parseElemWindowOnly)
+                .filter(Objects::nonNull);
   }
 
   @Override