diff --git a/sdks/python/apache_beam/yaml/yaml_ml.py b/sdks/python/apache_beam/yaml/yaml_ml.py
index 4e750b79ce30..51f18c733046 100644
--- a/sdks/python/apache_beam/yaml/yaml_ml.py
+++ b/sdks/python/apache_beam/yaml/yaml_ml.py
@@ -487,9 +487,10 @@ def fn(x: PredictionResult):
               model_handler_provider._postprocess_fn_internal()),
           inference_args=inference_args)
       | beam.Map(
-          lambda row: beam.Row(**{
-              inference_tag: row[1], **row[0]._asdict()
-          })).with_output_types(schema))
+          lambda row: beam.Row(
+              **{
+                  **row[0]._asdict(), str(inference_tag): row[1]
+              })).with_output_types(schema))
 
 
 def _config_to_obj(spec):
diff --git a/sdks/python/pyrefly.toml b/sdks/python/pyrefly.toml
index 3b6cc5e975ce..79fd75a46262 100644
--- a/sdks/python/pyrefly.toml
+++ b/sdks/python/pyrefly.toml
@@ -58,5 +58,4 @@ bad-context-manager = "ignore"
 invalid-yield = "ignore"
 bad-argument-count = "ignore"
 bad-typed-dict-key = "ignore"
-no-access = "ignore"
-bad-unpacking = "ignore"
\ No newline at end of file
+no-access = "ignore"
\ No newline at end of file