diff --git a/test/datasets_utils.py b/test/datasets_utils.py
index 337a7382366..4e3fd0ac0e3 100644
--- a/test/datasets_utils.py
+++ b/test/datasets_utils.py
@@ -572,7 +572,7 @@ def create_image_file(
 
     image = create_image_or_video_tensor(size)
     file = pathlib.Path(root) / name
-    PIL.Image.fromarray(image.permute(2, 1, 0).numpy()).save(file)
+    PIL.Image.fromarray(image.permute(2, 1, 0).numpy()).save(file, **kwargs)
     return file
 
 
@@ -708,6 +708,6 @@ def size(idx):
     os.makedirs(root)
 
     return [
-        create_video_file(root, file_name_fn(idx), size=size(idx) if callable(size) else size)
+        create_video_file(root, file_name_fn(idx), size=size(idx) if callable(size) else size, **kwargs)
         for idx in range(num_examples)
     ]