diff --git a/Liutex.py b/Liutex.py
index 7aee9c73acce2042019824f6b0cefd58e9f6fdeb..7bcb4d279883ec6d5c94d55c3022319fe1e9d51e 100644
--- a/Liutex.py
+++ b/Liutex.py
@@ -200,8 +200,8 @@ def main():
     mesh = update_mesh_with_omega_liutex(mesh, a, b)
 
     # Save the mesh with vorticity and omega data to a VTK file
-    mesh.save("Liu_Rortex_T106C_Re80K_19.09.vtk")
-    print("Dataset saved as Liu_Rortex_T106C_Re80K_19.09.vtk")
+    mesh.save("Liu_Rortex_T106C_Re80K_19.09.vtu")
+    print("Dataset saved as Liu_Rortex_T106C_Re80K_19.09.vtu")
 
     # Selecting seed points in high vorticity regions
     vorticity_threshold = np.percentile(vorticity_magnitude, 90)  # Top 10% vorticity magnitude
@@ -224,8 +224,8 @@ def main():
 
     # Save vorticity lines
     vorticity_mesh = mesh.streamlines_from_source(seed_polydata, vectors='Velocity')
-    vorticity_mesh.save("Vorticity_Lines_Blade_19.09.vtk")
-    print("Vorticity lines saved as Vorticity_Lines_Blade_19.09.vtk")
+    vorticity_mesh.save("Vorticity_Lines_Blade_19.09.vtu")
+    print("Vorticity lines saved as Vorticity_Lines_Blade_19.09.vtu")
 
     plotter.add_axes()
     plotter.view_isometric()