diff --git a/DocumentUnderstanding/VGT/object_detection/pdf2img.py b/DocumentUnderstanding/VGT/object_detection/pdf2img.py
index b73ad27..c16704e 100644
--- a/DocumentUnderstanding/VGT/object_detection/pdf2img.py
+++ b/DocumentUnderstanding/VGT/object_detection/pdf2img.py
@@ -34,6 +34,6 @@
     for i, image in enumerate(images):
         image.save(
             os.path.join(
-                args.output, f"page_{i}.png"))
+                args.output, f"page_{i}.{args.format}"))
 
     logging.info(f"PDF converted to images and saved at {args.output}")