diff --git a/sdks/java/core/src/main/java/org/apache/beam/sdk/values/TaggedPValue.java b/sdks/java/core/src/main/java/org/apache/beam/sdk/values/TaggedPValue.java
index ded5bef31131..bfd08202114d 100644
--- a/sdks/java/core/src/main/java/org/apache/beam/sdk/values/TaggedPValue.java
+++ b/sdks/java/core/src/main/java/org/apache/beam/sdk/values/TaggedPValue.java
@@ -19,6 +19,7 @@
 
 import com.google.auto.value.AutoValue;
 import org.apache.beam.sdk.annotations.Internal;
+import org.apache.beam.sdk.util.Preconditions;
 import org.apache.beam.vendor.guava.v32_1_2_jre.com.google.common.collect.Iterables;
 
 /**
@@ -33,9 +34,10 @@ public static TaggedPValue of(TupleTag<?> tag, PCollection<?> value) {
     return new AutoValue_TaggedPValue(tag, value);
   }
 
-  @SuppressWarnings({"keyfor", "nullness"})
   public static TaggedPValue ofExpandedValue(PCollection<?> value) {
-    return of(Iterables.getOnlyElement(value.expand().keySet()), value);
+    return of(
+        Preconditions.checkArgumentNotNull(Iterables.getOnlyElement(value.expand().keySet())),
+        value);
   }
 
   /** Returns the local tag associated with the {@link PValue}. */