diff --git a/python/scripts/convert_format.py b/python/scripts/convert_format.py
index 8f02b3e..ac2d804 100644
--- a/python/scripts/convert_format.py
+++ b/python/scripts/convert_format.py
@@ -80,13 +80,13 @@ def files_to_ply(input_dir, output_dir):
     if not os.path.exists(output_dir):
         os.makedirs(output_dir)
 
-    # Iterate through txt files in input directory
+    # Iterate through files in input directory
     for filename in os.listdir(input_dir):
         if filename.endswith('.las'):
             input_file = os.path.join(input_dir, filename)
             output_file = os.path.join(output_dir, filename.replace('.las', '.ply'))
             read_pc_np(input_file, output_file)
-        elif filename.endswith('.txt'):
+        elif filename.endswith('.laz'):
             input_file = os.path.join(input_dir, filename)
             output_file = os.path.join(output_dir, filename.replace('.laz', '.ply'))
             read_pc_np(input_file, output_file)
@@ -103,4 +103,4 @@ def files_to_ply(input_dir, output_dir):
 
     input_dir = args.input_dir
     output_dir = args.output_dir
-    files_to_ply(input_dir, output_dir)
\ No newline at end of file
+    files_to_ply(input_dir, output_dir)