diff --git a/dan/datasets/extract/extract.py b/dan/datasets/extract/extract.py
index 65a73488241ea0ecc1146dff46316265d50bcd8c..c00bd6968a7d996a0856f6032bb12e0fc580acea 100644
--- a/dan/datasets/extract/extract.py
+++ b/dan/datasets/extract/extract.py
@@ -330,9 +330,9 @@ def run(
     entity_worker_version: Optional[Union[str, bool]],
     max_width: Optional[int],
     max_height: Optional[int],
+    image_format: str,
     cache_dir: Path,
     keep_spaces: bool,
-    img_format: str,
 ):
     assert database.exists(), f"No file found @ {database}"
     open_database(path=database)
@@ -360,5 +360,5 @@ def run(
         max_height=max_height,
         cache_dir=cache_dir,
         keep_spaces=keep_spaces,
-        image_extension=img_format,
+        image_extension=image_format,
     ).run()