diff --git a/utils.py b/utils.py
index cafe31d..9bc7788 100644
--- a/utils.py
+++ b/utils.py
@@ -53,7 +53,8 @@ def tensor2pil(image: torch.Tensor) -> List[Image.Image]:
 
     if batch_count > 1:
         out = []
-        out.extend([tensor2pil(image[i]) for i in range(batch_count)])
+        for i in range(batch_count):
+            out.extend(tensor2pil(image[i]))
         return out
 
     return [