diff --git a/egs/wenetspeech/ASR/local/text2segments.py b/egs/wenetspeech/ASR/local/text2segments.py
index acf6f9698..3df727c67 100644
--- a/egs/wenetspeech/ASR/local/text2segments.py
+++ b/egs/wenetspeech/ASR/local/text2segments.py
@@ -61,8 +61,8 @@ def main():
     parser = get_parser()
     args = parser.parse_args()
 
-    input_file = args.input
-    output_file = args.output
+    input_file = args.input_file
+    output_file = args.output_file
 
     f = open(input_file, "r", encoding="utf-8")
     lines = f.readlines()