diff --git a/data_generator.py b/data_generator.py
index 28735a9..87a4518 100644
--- a/data_generator.py
+++ b/data_generator.py
@@ -218,8 +218,6 @@ class DataGenerator:
                 file_path = os.path.join(depth_dir, depth_file)
                 new_file_path = os.path.join(depth_dir, f"{name}.png")
                 os.rename(file_path,new_file_path)
-        BlenderUtils.save_blend(scene_dir)
-        exit(0)
         
         return True
     
diff --git a/utils/blender_util.py b/utils/blender_util.py
index 080670c..a6608f6 100644
--- a/utils/blender_util.py
+++ b/utils/blender_util.py
@@ -279,6 +279,7 @@ class BlenderUtils:
             scene.render.filepath = os.path.join(
                 output_dir, mask_dir, f"{file_name}_{cam_suffix}.png"
             )
+            scene.render.image_settings.file_format = "PNG"
             scene.render.image_settings.color_depth = "8"
             scene.render.resolution_percentage = 100
             scene.render.use_overwrite = False