From 919e8a4bd5826bcd5c3671793517a6227d4a8154 Mon Sep 17 00:00:00 2001
From: Nick Li <56149585+nickuncaged1201@users.noreply.github.com>
Date: Wed, 10 May 2023 17:37:40 -0700
Subject: [PATCH] fixup (#26657)

---
 .../FileWriteSchemaTransformProvider.java      | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

diff --git a/sdks/java/io/file-schema-transform/src/main/java/org/apache/beam/sdk/io/fileschematransform/FileWriteSchemaTransformProvider.java b/sdks/java/io/file-schema-transform/src/main/java/org/apache/beam/sdk/io/fileschematransform/FileWriteSchemaTransformProvider.java
index b3550e7dbb154..f0070f13919b4 100644
--- a/sdks/java/io/file-schema-transform/src/main/java/org/apache/beam/sdk/io/fileschematransform/FileWriteSchemaTransformProvider.java
+++ b/sdks/java/io/file-schema-transform/src/main/java/org/apache/beam/sdk/io/fileschematransform/FileWriteSchemaTransformProvider.java
@@ -118,14 +118,16 @@ public PCollectionRowTuple expand(PCollectionRowTuple input) {
 
       PCollection<String> files = rowInput.apply("Write Rows", transform);
       PCollection<Row> output =
-          files.apply(
-              "Filenames to Rows",
-              MapElements.into(rows())
-                  .via(
-                      (String name) ->
-                          Row.withSchema(OUTPUT_SCHEMA)
-                              .withFieldValue(FILE_NAME_FIELD.getName(), name)
-                              .build()));
+          files
+              .apply(
+                  "Filenames to Rows",
+                  MapElements.into(rows())
+                      .via(
+                          (String name) ->
+                              Row.withSchema(OUTPUT_SCHEMA)
+                                  .withFieldValue(FILE_NAME_FIELD.getName(), name)
+                                  .build()))
+              .setRowSchema(OUTPUT_SCHEMA);
 
       return PCollectionRowTuple.of(OUTPUT_TAG, output);
     }