diff --git a/sdk/python/kfp/compiler/_component_builder.py b/sdk/python/kfp/compiler/_component_builder.py
index b1bd63f214f1..d176bdb30aca 100644
--- a/sdk/python/kfp/compiler/_component_builder.py
+++ b/sdk/python/kfp/compiler/_component_builder.py
@@ -339,9 +339,11 @@ def _generate_entrypoint(self, component_func, python_version='python3'):
     func_signature = 'def ' + new_func_name + '('
     for input_arg in input_args:
       func_signature += input_arg + ','
-    func_signature += '_output_file):'
     if output_is_named_tuple:
-      func_signature = func_signature.replace('file', 'files')
+      func_signature += '_output_files'
+    else:
+      func_signature += '_output_file'
+    func_signature += '):'
     codegen.writeline(func_signature)
 
     # Call user function